# MT-YOLOv6 [About Naming YOLOv6](./docs/About_naming_yolov6.md)
## Introduction
YOLOv6 is a single-stage object detection framework dedicated to industrial applications, with hardware-friendly efficient design and high performance.
<img src="assets/picture.png" width="800">
YOLOv6-nano achieves 35.0 mAP on COCO val2017 dataset with 1242 FPS on T4 using TensorRT FP16 for bs32 inference, and YOLOv6-s achieves 43.1 mAP on COCO val2017 dataset with 520 FPS on T4 using TensorRT FP16 for bs32 inference.
YOLOv6 is composed of the following methods:
- Hardware-friendly Design for Backbone and Neck
- Efficient Decoupled Head with SIoU Loss
## Coming soon
- [ ] YOLOv6 m/l/x model.
- [ ] Deployment for MNN/TNN/NCNN/CoreML...
- [ ] Quantization tools
## Quick Start
### Install
```shell
git clone https://github.com/meituan/YOLOv6
cd YOLOv6
pip install -r requirements.txt
```
### Inference
First, download a pretrained model from the YOLOv6 [release](https://github.com/meituan/YOLOv6/releases/tag/0.1.0)
Second, run inference with `tools/infer.py`
```shell
python tools/infer.py --weights yolov6s.pt --source img.jpg / imgdir
yolov6n.pt
```
### Training
Single GPU
```shell
python tools/train.py --batch 32 --conf configs/yolov6s.py --data data/coco.yaml --device 0
configs/yolov6n.py
```
Multi GPUs (DDP mode recommended)
```shell
python -m torch.distributed.launch --nproc_per_node 8 tools/train.py --batch 256 --conf configs/yolov6s.py --data data/coco.yaml --device 0,1,2,3,4,5,6,7
configs/yolov6n.py
```
- conf: select config file to specify network/optimizer/hyperparameters
- data: prepare [COCO](http://cocodataset.org) dataset and specify dataset paths in data.yaml
### Evaluation
Reproduce mAP on COCO val2017 dataset
```shell
python tools/eval.py --data data/coco.yaml --batch 32 --weights yolov6s.pt --task val
yolov6n.pt
```
### Deployment
* [ONNX](./deploy/ONNX)
* [OpenVINO](./deploy/OpenVINO)
### Tutorials
* [Train custom data](./docs/Train_custom_data.md)
* [Test speed](./docs/Test_speed.md)
## Benchmark
| Model | Size | mAP<sup>val<br/>0.5:0.95 | Speed<sup>V100<br/>fp16 b32 <br/>(ms) | Speed<sup>V100<br/>fp32 b32 <br/>(ms) | Speed<sup>T4<br/>trt fp16 b1 <br/>(fps) | Speed<sup>T4<br/>trt fp16 b32 <br/>(fps) | Params<br/><sup> (M) | Flops<br/><sup> (G) |
| :-------------- | ----------- | :----------------------- | :------------------------------------ | :------------------------------------ | ---------------------------------------- | ----------------------------------------- | --------------- | -------------- |
| [**YOLOv6-n**](https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6n.pt) | 416<br/>640 | 30.8<br/>35.0 | 0.3<br/>0.5 | 0.4<br/>0.7 | 1100<br/>788 | 2716<br/>1242 | 4.3<br/>4.3 | 4.7<br/>11.1 |
| [**YOLOv6-tiny**](https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6t.pt) | 640 | 41.3 | 0.9 | 1.5 | 425 | 602 | 15.0 | 36.7 |
| [**YOLOv6-s**](https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6s.pt) | 640 | 43.1 | 1.0 | 1.7 | 373 | 520 | 17.2 | 44.2 |
- Comparisons of the mAP and speed of different object detectors are tested on [COCO val2017](https://cocodataset.org/#download) dataset.
- Refer to [Test speed](./docs/Test_speed.md) tutorial to reproduce the speed results of YOLOv6.
- Params and Flops of YOLOv6 are estimated on deployed model.
- Speed results of other methods are tested in our environment using official codebase and model if not found from the corresponding official release.
## Third-party resources
* YOLOv6 NCNN Android app demo: [ncnn-android-yolov6](https://github.com/FeiGeChuanShu/ncnn-android-yolov6) from [FeiGeChuanShu](https://github.com/FeiGeChuanShu)
* YOLOv6 ONNXRuntime/MNN/TNN C++: [YOLOv6-ORT](https://github.com/DefTruth/lite.ai.toolkit/blob/main/lite/ort/cv/yolov6.cpp), [YOLOv6-MNN](https://github.com/DefTruth/lite.ai.toolkit/blob/main/lite/mnn/cv/mnn_yolov6.cpp) and [YOLOv6-TNN](https://github.com/DefTruth/lite.ai.toolkit/blob/main/lite/tnn/cv/tnn_yolov6.cpp) from [DefTruth](https://github.com/DefTruth)
* YOLOv6 TensorRT Python: [yolov6-tensorrt-python](https://github.com/Linaom1214/tensorrt-python/blob/main/yolov6/trt.py) from [Linaom1214](https://github.com/Linaom1214)
* YOLOv6 TensorRT Windows C++: [yolort](https://github.com/zhiqwang/yolov5-rt-stack/tree/main/deployment/tensorrt-yolov6) from [Wei Zeng](https://github.com/Wulingtian)
没有合适的资源?快使用搜索试试~ 我知道了~
资源详情
资源评论
资源推荐
收起资源包目录
YOLOv6-main.zip (60个子文件)
YOLOv6-main
docs
Train_custom_data.md 4KB
About_naming_yolov6.md 767B
Test_speed.md 1KB
deploy
OpenVINO
export_openvino.py 4KB
README.md 667B
ONNX
README.md 3KB
export_onnx.py 5KB
tools
eval.py 3KB
train.py 4KB
quantization
tensorrt
training_aware
QAT_quantizer.py 2KB
post_training
quant.sh 733B
LICENSE 11KB
README.md 4KB
Calibrator.py 9KB
onnx_to_tensorrt.py 10KB
requirements.txt 147B
mnn
README.md 14B
infer.py 4KB
data
coco.yaml 1KB
images
image2.jpg 140KB
image1.jpg 79KB
image3.jpg 115KB
dataset.yaml 596B
assets
picture.png 517KB
LICENSE 34KB
configs
yolov6s_finetune.py 1KB
yolov6_tiny.py 1KB
yolov6s.py 1KB
yolov6n.py 1KB
yolov6_tiny_finetune.py 1KB
yolov6n_finetune.py 1KB
requirements.txt 311B
.gitignore 1KB
yolov6
models
efficientrep.py 3KB
loss.py 16KB
yolo.py 3KB
reppan.py 3KB
effidehead.py 7KB
end2end.py 6KB
core
evaler.py 11KB
engine.py 12KB
inferer.py 8KB
data
data_augment.py 8KB
datasets.py 20KB
data_load.py 3KB
solver
build.py 1KB
layers
common.py 21KB
dbb_transforms.py 2KB
utils
envs.py 1KB
Arial.ttf 755KB
events.py 1KB
nms.py 5KB
general.py 421B
torch_utils.py 3KB
figure_iou.py 5KB
checkpoint.py 2KB
config.py 3KB
ema.py 2KB
README.md 5KB
.pre-commit-config.yaml 176B
共 60 条
- 1
小凹兔
- 粉丝: 2713
- 资源: 14
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- UI Particle Image v1.2.2.unitypackage
- Highlight Plus All in One Outline Selection Effects v12.0
- ALINE v1.7.4.unitypackage
- Curvy Splines 8 v8.9.0.unitypackage
- Touch Camera PRO 2024 Easy Mobile Desktop Camera Controller3.1.2
- Translucent Image - Fast UI Background Blur v3.16.0.unitypackage
- STM32F103RC串口收发DMA程序
- 环境部署配置脚本-Tomcat
- SpringBoot+Vue实现图片滑块和文字点击验证码
- 基于参数优化多样性熵的滚动轴承故障诊断源代码,实现了对多个数据集的测试,并添加了排列熵、样本熵等方法的对比
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论0