源码地址:https://github.com/smallcorgi/Faster-RCNN_TF
我自己做的数据集和voc的格式一样,所以只修改classes 里的类别就好。
1、/Faster-RCNN_TF/lib/datasets/pascal_voc.py
self._classes 修改为自己数据的类
2、/Faster-RCNN_TF/lib/networks/VGGnet_train.py 和 /Faster-RCNN_TF/lib/networks/VGGnet_test.py
修改类的个数:n_classes为自己的类的个数+1
训练前要把官方的数据替换掉,如果你拿原始VOC数据训练过,还需要把load的数据和模型删掉。
需要删除这两个目录下的文件:(每次重新训练这两个文件最好都先删除掉)
/Faster-RCNN_TF/data/cache
/Faster-RCNN_TF/output/faster_rcnn_end2end