SSD训练自己的数据集

参考链接
https://www.jianshu.com/p/04ea90ebfd95
https://blog.youkuaiyun.com/dingkm666/article/details/88775428
https://blog.youkuaiyun.com/wenyunick/article/details/90203486?utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2_allfirst_rank_v2~rank_v25-1-90203486.nonecase&utm_term=pytorch%E5%AE%9E%E7%8E%B0ssd%E8%87%AA%E5%B7%B1%E7%9A%84%E6%95%B0%E6%8D%AE%E9%9B%86

修改mutibox_loss.py
第114行
原来为N = num_pos.data.sum()
参考文章建议改为N = num_pos.data.sum().double()
这里不必修改，修改后报错

我们检测的只有一类需要，特别注意，需要加[]
VOC_CLASSES = [(# always index 0
‘ship’)]