1、制作数据放在data/VOCdevkit2007/VOC2007中
2、修改模型
"/home/pengshengfeng/py-faster-
rcnn/models/pascal_voc/VGG16/faster_rcnn_end2end/train.prototxt"
1)num_classes :, 则为3(n + 背景类)
2)cls_score 层, num_output :也设置为3(n + 背景类)
3)bbox_pred层, num_output:设置为12(n + 背景类) ×4
4)name: 'roi-data'层 num_classes 3
对于test.prototxt文件, 修改如下:
1)cls_score 层,num_output :也设置为3(n + 背景类)
2)bbox_pred层, num_output:设置为12(n + 背景类) ×4
3,修改数据输入类别:cd /lib/datasets
首先,修改pascal_voc.py,
self.CLASS = (
‘background‘, ‘nescafe’,'max well'
。。。
)
4、解决错误:
rcnn/tools/../lib/datasets/imdb