Skip to content

alir14/yolov5-opencv-dnn-cpp

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

40 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

yolov5-opencv-dnn-cpp

使用opencv模块部署yolov5-6.0版本

基于6.0版本的yolov5:https://github.com/ultralytics/yolov5

OpenCV>=4.5.0

$ python path/to/export.py --weights yolov5s.pt --img [640,640] --opset 12 --include onnx

2022.12.13 更新:

  • 如果你的显卡支持FP16推理的话,可以将模型读取代码中的DNN_TARGET_CUDA改成DNN_TARGET_CUDA_FP16提升推理速度(虽然是蚊子腿,好歹也是肉(:

2022.03.29 更新:

  • 新增P6模型支持,可以通过yolo.h中定义的YOLO_P6切换

  • 另外关于换行符,windows下面需要设置为CRLF,上传到github会自动切换成LF,windows下面切换一下即可

以下图片为更新p6模型之后yolov5s6.onnx运行结果: zidane

bus

About

Deploy yolov5-6.0 version using opencv module

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 100.0%