Faster R-CNN源码网址:https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5点击打开链接
(电脑的相关配置可根据源码网址中进行,本环境下已经配置完成)
一:数据集
根据https://blog.csdn.net/weixin_38124357/article/details/78425890制作好数据集后放置在路径“Faster-RCNN-TensorFlow-Python3.5-master\data\VOCdevkit2007\VOC2007“对应的文件中。
二:更改类别
1、训练前:
Faster-RCNN-TensorFlow-Python3.5-master\lib\datasets\pascal_voc.py文件中,将以下代码中的类别替换为自己数据集的类别(不要删除类别!!不要更改'__background__'!!例如我需要分类的类别为card1,则选择一个原有类别更换为“card1”)
self._classes = ('__background__', # always index 0
'card1', 'bicycle', 'bird', 'boat',