YOLOV4-Tiny:You Only Look Once-Tiny目标检测模型在TF2当中的实现
2021年2月7日更新:
仔细对照了darknet库的网络结构,发现P5_Upsample和feat1的顺序搞反了,已经调整,重新训练了权值,加入letterbox_image的选项,关闭letterbox_image后网络的map得到提升。
目录
性能情况
训练数据集
权值文件名称
测试数据集
输入图片大小
mAP 0.5:0.95
mAP 0.5
VOC07+12+COCO
VOC-Test07
416x416
-
77.5
COCO-Train2017
COCO-Val2017
416x416
21.8
41.3
所需环境
tensorflow-gpu==2.2.0
注意事项
代码中的yolov4_tiny_weights_coco.h5和yolov4_tiny_weights_vo
1