使用YOLOV4 轻松进行开关开闭检测
最近接到了一个开关开闭状态检测的项目,准确率要求很高,要达到99.9%以上,所以顺道试验一下yolov4的性能
yoloV4的训练过程跟yolov3训练并无差异,我把yolov4的git地址和yolov3训练的参考的博客地址放在这里
训练yolov3的过程
yolov4的代码地址
如果需要yolov4的权重文件和主干网络的权重文件,请在这里下载
我训练集使用了450张图片,测试效果时使用了1000个图片
训练结束后,跟以前一样用opencv直接调用cfg文件和权重文件发现报错了,无法直接调用
这里我使用的是pytorch调用darknet的方式来调用模型进行测试:
测试算法代码的地址在这里:
https://github.com/gitTongzhen/pytorch_yolov4_detect
yolov4的准确率满足要求,在GPU的实时性还行,大致为1080TI的服务器50ms一张图片,但是单位过于贫穷,没有gpu的服务器,项目需要部署到CPU端,实时性要求比较高,目前测试的i5双核的每张图片需要5000ms,和现在还不理解为什么cpu端和gpu端的性能会差这么多,有大佬看到的话,希望能够赐教。
目前只能放弃yolov4部署到cpu端的想法
效果图如下图所示


本文介绍使用YOLOV4进行开关开闭状态检测的项目经验,包括模型训练过程及遇到的问题。该项目准确率达到99.9%,但在CPU端部署时遇到性能瓶颈。
578





