YOLOv6 4.0 使用记录： OpenCV DNN C++推理_c++ opencv yolo 摄像头

作者：你好赵伟 | 2024-04-06 11:01:44

踩

c++ opencv yolo 摄像头

下载最新的4.0版本的

我下的是YOLOv6Lite-S

cd到项目目录，运行

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

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

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

导出报错。。。。：

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

成功获得onnx格式的文件

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

yolov5n 推理的话 150ms左右

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

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

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

参考：

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/371513