目录

1、下载源码

 2、下载权重文件

 3、配置环境

4、推理

6、ONNX格式导出

权重文件为yolov6list_s.pt

权重为yolov6.pt

 7、opencv DNN推理

8、个人总结

---

1、下载源码

下载最新的4.0版本的

 2、下载权重文件

我下的是YOLOv6Lite-S

 3、配置环境

cd到项目目录，运行

 pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

4、推理

图片推理没问题，但是输入为摄像头的时候不会弹出显示窗口

 python tools/infer.py  --weights yolov6lite_s.pt  --source data/images/image1.jpg 

6、ONNX格式导出

权重文件为yolov6list_s.pt

导出报错。。。。：

AttributeError: 'ConvModule' object has no attribute 'act'

权重为yolov6.pt

成功获得onnx格式的文件

 7、opencv DNN推理

我直接用摄像头，推理代码用的是之前写的博客代码，输入应该是设置为640，CPU推理耗时120ms左右。

yolov5n 推理的话 150ms左右

8、个人总结

yolov6n 推理速度确实比yolov5n快一点，检测的物体更多（平放的书本也检测出来了），但是存在一定的误检，可能修改置信度阈值能好点？

 但是我看 ONNX/export_onnx.py的置信度阈值设置还可以。暂时不知道如何改进，如有大佬知道，烦请留言

觉得写得不错，厚脸皮要个赞，哈哈哈

参考：

https://yolov6-docs.readthedocs.io/zh_CN/latest/%E9%83%A8%E7%BD%B2/ONNX.html

手把手教你运行YOLOv6（超详细）_Mr Dinosaur的博客-CSDN博客