ssd目标检测训练自己的数据_使用SSD训练自己的数据+源码(pytorch)学习记录(一)

最近的一个任务用到了SSD作检测，将这份代码应用到了自己的数据上，为了防止自己忘了，其中的整个过程作一个记录，同时这两天决定将SSD的源码好好读一遍，加深一下理解。所用的代码地址：

https://github.com/lufficc/SSD​github.com

这份代码是今年的，比起两年前star数量最多的SSD Pytorch实现有更多的灵活度，更详细的文档，作者给出了不同的Backbone，在readme里面给出了特别详细的修改指导，对于想要自己更改网络，写自己的数据类的小伙伴，强烈安利！在此感谢原作者！

因为安装使用SSD和源码解读是很大两块内容，分俩个部分来写，第一部分主要是关于SSD的使用。

SSD安装/训练/推理

安装：（ubuntu环境）对于如何安装并跑这份代码作者在readme里面写得简直不能太详尽，只补充几个注意点：

• 养成好习惯，尽量在conda 里面创个虚拟环境
• CUDA9/10皆可，但是CUDA8是不行的
• 小Tip: conda install torchvision==0.3.0，一行命令基本帮忙把torch相关依赖都装上了

插播一下数据准备：

新建一个datasets文件夹。源码支持coco/voc格式，当然你也可以自定义一个数据接口，这个作者也有指导！自己写数据类请参考ssd/data/datasets/目录下的coco.py和voc.py

对于voc:

--datasets
|