目录
1.准备自己的数据集
pytorch-ssd可以支持VOC和COCO两种数据格式,所以你要将自己的数据集转化为VOC格式或者COCO格式,相应的转化脚本,请点击(注:SSD-Pytorch中含有自动下载VOC和COCO官方数据的脚本)如下图
2.pytorch环境安装即SSD-pytorch代码下载
点击链接登陆pytorch官网根据电脑的操作系统配置选择安装: PyTorch
python版本要求python3 + :Welcome to Python.org
SSD-Pytorch代码链接:GitHub - amdegroot/ssd.pytorch: A PyTorch Implementation of Single Shot MultiBox Detector
训练模型下载链接(官方模型链接似乎失效了,推荐下载):https://download.csdn.net/download/qq_34374211/10712378
3.正常的修改
根据经验,我们要清楚自己数据集的含有的类别,例如:我自己的数据集为VOC格式,含有8个类别。