YOLOV4:You Only Look Once目标检测模型-修改mobilenet系列主干网络-在pytorch当中的实现
2021年2月8日更新:
加入letterbox_image的选项,关闭letterbox_image后网络的map一般可以得到提升。
目录
性能情况
训练数据集
权值文件名称
测试数据集
输入图片大小
mAP 0.5:0.95
mAP 0.5
VOC07+12
VOC-Test07
416x416
-
79.72
VOC07+12
VOC-Test07
416x416
-
80.12
VOC07+12
VOC-Test07
416x416
-
79.01
所需环境
torch==1.2.0
注意事项
提供的三个权重分别是基于mobilenetv1、mobilenetv2、mobilenetv3主干网络训练而成的。使用的时候注意backbone和权重的对应。
训练
1