1、将数据集和标签文件放到vocdevikt文件夹下的文件夹里(对应放入)
Annotations和JPEGImages放.xml和原数据集图片,建立ImageSets文件夹里建立Main放四个空文档。
注意:标签文件和图片文件的名字一定要一样!!!
尤其是.xml中的filename(文件名字),后缀也要一样!!!
不然会报错:
2、将权重文件压缩包放到src文件夹下
3、在.json文件里改类名
smoke后应该是2
依次类推后面是3、4、5等
4、Train_ssd300.py里的creat_model函数改类数(需用类数+1,背景类)
5、运行test.py文件,划分数据集里的测试集训练集验证集
6、运行train_ssd300.py文件
原类:
Pycharm 不清空控制台运行下一个程序的做法: