用opencv的dnn模块实现Yolo-Fastest的目标检测

本文介绍如何使用OpenCV的DNN模块进行Yolo-Fastest目标检测,并分享了作者自己编写的Python程序及所需版本的OpenCV。此外,还提供了YOLOv3、YOLOv4等其他版本的实现。

最近在微信朋友圈里疯传的Yolo-Fastest(github代码地址是https://github.com/dog-qiuqiu/Yolo-Fastest),据说模型非常小,且运行速度快,而且,这个YOLO-Fastest算法满足所有平台的需要。不论是PyTorch,还是Tensorflow,又或者是Keras和Caffe,可以说是全平台通用。但是我没有看到用opencv的dnn模块做前向推理的,于是我就想编写一个程序用opencv的dnn模块做Yolo-Fastest的前向推理。而在此之前,就已经有用opencv的dnn模块做yolov3目标检测(github代码地址是https://github.com/spmallick/learnopencv/blob/master/ObjectDetection-YOLO/object_detection_yolo.py),那么很自然的一个想法是下载Yolo-Fastest的源码,拷贝cfg和weight文件,然后修改object_detection_yolo.py里的cfg和weight文件路径为Yolo-Fastest里的cfg和weight文件,然而运行程序,结果ok。注意,运行这个程序,opencv需要是4.0版本以上的,我之前的opencv版本是3.4的,运行这个程序直接报错中断了,输入pip install --upgrade opencv-python,把opencv更新到最新的opencv4.4,重新运行程序就正常了。

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值