YoloV4学习笔记5_yolov4-tiny(win10+VC2015+opencv4.4+yolov4)

YoloV4学习笔记5_yolov4-tiny(win10+VC2015+opencv4.4+yolov4)

对于cpu识别或者资源不太充足的设备,yolov4的完整版太慢了。
yolov4-tiny相当于yolov4的缩小版本,速度比完整版快10倍,当然也牺牲了一些准确度。
》YOLOv4-tiny is a smaller/faster version of YOLOv4.
》And YOLOv4-tiny-3l is a 3-layer version which is somewhere between YOLOv4 and YOLOv4-tiny.
》YOLOv4-tiny-3l is better at finding small objects; “3l” refers to 3 YOLO layers vs the usual 2 in “tiny”

yolov4-tiny在特征提取时没有采用Mish**函数,并且在特征加强层只采用了一个特征金字塔,没有像yolov4那样再进行下采样。
yolov4的训练参数有64363101个,而yolov4-tiny只有5918006个,比yolov4少了十倍。

yolov4-tiny只选择了两层来送入特征金字塔。所以对于小物体检测,以及两个目标叠加在一起(一个目标将另一个目标遮挡了一部分,如图一)是很难检测到的。其它的对于独立的物体检测的效果感觉还是不错的,如图二。
YoloV4学习笔记5_yolov4-tiny(win10+VC2015+opencv4.4+yolov4)_第1张图片
YoloV4学习笔记5_yolov4-tiny(win10+VC2015+opencv4.4+yolov4)_第2张图片

训练:
yolov4-tiny和完整版的使用训练方法一样,只是cfg配置文件和weights预训练权重文件不同。
下载地址:
https://github.com/AlexeyAB/darknet#how-to-train-tiny-yolo-to-detect-your-custom-objects

修改cfg配置文件,修改方法参考
https://editor.csdn.net/md/?articleId=125670796

你可能感兴趣的:(深度学习,VC,学习,人工智能)