Faster R-CNN(Tensorflow版) 训练自己的数据集

源码地址: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

参考文章:https://www.jianshu.com/p/c0c239a67f8e

猜你喜欢

转载自blog.csdn.net/ytusdc/article/details/80406641
今日推荐