MobileNetV3-SSD-Compact-Version
MobileNetV3 SSD的简洁版本
希望尝试新技术请到这里
一个轻量级的目标检测包括多种模型
环境
Ubuntu18.04
版本 PyTorch 1.4
如果您想从头开始需要使用
mbv3_large.old.pth.tar
是backbone用来参数初始化的模型
有可能会历经坎坷
简便方式就是使用预训练模型
如果您要直接测试,模型改名为checkpoint_ssd300.pth.tar
模型下载地址
链接:
提取码:gwwv
模型测试结果 mAP 0.679 (未在COCO数据集做预训练版本)
使用步骤
一 下载VOC数据集之后,将VOCtrainval_06-Nov-2007和VOCtest_06-Nov-2007合并在一起
数据集下载 可以看这里
二 先打开create_data_lists.py文件
改成自己数
1