Mobilenet-YOLO-Pytorch
就像我之前的项目,损失函数与原始实现非常相似
模型
pytorch实现的MobileNet-YOLO检测网络,在07 + 12上进行了训练,在VOC2007上进行了测试(图像网络经过预训练,而不是coco)
网络
地图
解析度
yolov3
约洛夫4
MobileNetV2
71.2
352
✓
MobileNetV2
352
✓
MobileNetV3
MobileNetV3
训练步骤
下载数据集VOCdevkit /,如果已有,请跳过此步骤
sh scripts/VOC2007.sh
sh scripts/VOC2012.sh
创建lmdb
sh scripts/create.sh
开始训练
sh scripts/train.sh
演示版
下载 ,并保存在$ Mobilenet-YOLO-Pytorch / chec
1