Pytorch基于FSSD训练自己数据集+wins10

Pytorch基于FSSD训练自己数据集
1、下载FSSD代码:
https://github.com/dlyldxwl/fssd.pytorch

2、在data中放入自己整理好的数据集如下:
Pytorch基于FSSD训练自己数据集+wins10_第1张图片
3、创建weights文件夹,并下载训练权重文件()
4、修改data/voc0712.py文件中VOC_CLASS修改成自己的类型
5、修改FSSD——VGG.py文件中的build_net的num_class修改为自己的种类(+背景)
6、因为pytorch版本问题,把VGG中的RELU(inplace)进行修改:

RELU(inplace=True

修改为

RELU(inplace=False

7、选择用VOC格式,就把相关COCO内容进行注释:
在train.py文件中:
8、在这里插入图片描述
Pytorch基于FSSD训练自己数据集+wins10_第2张图片
在data/init.py文件中:
Pytorch基于FSSD训练自己数据集+wins10_第3张图片
8、训练大功告成!!

慢慢来,你就成功了!
改代码错误不要慌,你就赢了!

你可能感兴趣的:(经典目标检测算法运行错误,python)