Yolo v4系列学习(五)Yolov4-tiny

        这几天,Alexey Bochkovskiy大神在Yolov4官网上更新了Yolov4-tiny!这对于计算资源相对紧张的开发者来说当然是福音啊。

这样的话,YOLO算法各个版本都有其对应的tiny版本了!

1、从如下网址下载最新版的darknetAB源码:

https://github.com/AlexeyAB/darknet

 2、下载 Yolov4-tiny训练权重:yolov4-tiny权重,将下载好的权重放到darknet目录下即可;

如果需要用Yolov4-tiny来进行训练,则需要下载Yolov4-tiny的预训练权重:yolov4-tiny预训练权重,具体的训练方法参考这里

3、打开makefile文件,在编译之前可以修改可选参数以适应于自己的计算机:

        (1)GPU=1,采用GPU加速,采用CUDA进行编译;
        (2)CUDNN=1,采用cuDNN v5-v7进行编译,进行加速训练;
        (3)CUDNN_HALF=1,在Titan V、Tesla V100、DGX-2及更高版本上使用;
        (4)OPENCV=1,使用OpenCV,支持各个版本的OpenCV(4.x/3.x/2.4.x),可以检测来自网络摄像机或web-cams的视频文件与视频流。

        上述这4个参数是主要修改的,其他参数使用得较少,可暂时不用修改。

4、在darknet目录下执行make进行编译;

5、执行如下指令进行图片检测:

./darknet detector test ./cfg/coco.data cfg/yolov4-tiny.cfg yolov4-tiny.weights data/dog.jpg

        检测结果如下:

 目标的置信度如下:

Done! Loaded 38 layers from weights-file 
data/dog.jpg: Predicted in 2.609000 milli-seconds.
bicycle: 29%
dog: 72%
truck: 82%
car: 46%

下面是Yolo_v4的检测结果:

Done! Loaded 162 layers from weights-file 
data/dog.jpg: Predicted in 27.039000 milli-seconds.
bicycle: 92%
dog: 98%
truck: 92%
pottedplant: 33%

        可以看到Yolov4-tiny检测准确度有下降,但是从耗时来看Yolov4-tiny优势明显:Yolov4-tiny检测的耗时仅为2.6毫秒,而Yolov4检测的耗时则为27毫秒,快了10倍有余!

猜你喜欢

转载自blog.csdn.net/JIEJINQUANIL/article/details/106998409
今日推荐