用faster RCNN训练自己的数据时候遇到的问题,具体为使用vgg16的end_2_end训练
这种问题一般都是train.prototxt的类别数没有改好导致的
在的vgg16的end_2_end下的train.prototxt需要改的是
data层的num_classes: n (自己要训练的类别+1,1代表背景)
以及 roi_data层的num_classes:n
cls_score层的num_output:n
bbox_preda层的num_output : 4*n
有问题欢迎交流