目录
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文章来源:https://www.toymoban.com/news/detail-433663.html
手把手教你运行YOLOv6(超详细)_Mr Dinosaur的博客-CSDN博客文章来源地址https://www.toymoban.com/news/detail-433663.html
到了这里,关于YOLOv6 4.0 使用记录: OpenCV DNN C++推理的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!