nqanh / affordance-net

AffordanceNet - Multiclass Instance Segmentation Framework - ICRA 2018
Other
121 stars 54 forks source link

How to change number of classes #42

Closed NilsDengler closed 4 years ago

NilsDengler commented 4 years ago

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?