smooth

xiaoxiao2021-02-28  104

用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

有问题欢迎交流

转载请注明原文地址: https://www.6miu.com/read-33067.html

最新回复(0)