SSD训练自己的数据时ssd_pascal.py需要更改的地方

先把所有的VOC0712替换成你的数据集名字

1、82、84行训练和测试数据所在的目录，train_data和test_data;

2、86、87行根据你想使用的尺寸选择是 300还是512；

3、 232行那个base_lr如果在你训练的时候loss出现Nan 就把他变小 ×0.1；

4、 237-263 检查路径是否正确；

5、 266行那个num_classes = 类别数目+1；

6、 332行 gpus 改成你想使用的显卡的id 可以设置多个；

7、 337行和338行的batch_size和accum_batch_size根据自己显卡情况更改；

      含义是每一次反向传播计算accum_batch_size个样本，batch_size是显卡每次计算的数目。

8、 359行 num_test_image 改成你测试样本的数目；

9、 367-391行中的数据根据你的需求更改，仔细更改，主要有step_value、max_iter、test_interval等