pytorch SSD项目目录

VOCdevkit：存放训练数据

weights ：存放权重文件

Config.py ：默认的一些配置

Test.py ：测试单张照片的识别

Train.py ：训练的py文件

augmentation.py：data augmentation的py文件，主要功能是扩大训练数据

detection.py：对识别的结果的数据进行部分筛选，传送给Test.py文件，供其调用使用

l2norm.py：进行l2正则化

loss_function.py：计算损失函数

ssd_net_vgg.py：ssd模型的实现

utils.py：工具类

voc0712.py：重写dataset类，提取voc的数据并规则化

主要是记录一下自己平时遇到的问题，和大家分享一下
如有侵犯，请联系我

点个赞支持一下吧