nqanh/affordance-net

How to change number of classes

Closed this issue · 0 comments

Hello,
I would like to train the net with my own data in pascal_voc format. Unfortunately it gives me the following error when I try to change the num_classes parameter in models/pascal_voc/VGG16/faster_rcnn_end2end/train.prototxt line 13 and 528:

Check failed: bottom[0]->channels() == bottom[1]->channels() (44 vs. 12)

What do I have to do to train the network with my number of classes?