search

rbgirshick / py-faster-rcnn

Faster R-CNN (Python implementation) -- see https://github.com/ShaoqingRen/faster_rcnn for the official MATLAB version
Other
8.11k stars 4.11k forks source link

Train with 2 classes #116

Open jcampoy69 opened 8 years ago

jcampoy69 commented 8 years ago

Hi,

Is there any docs on how to train a faster rcnn with only 2 classes?

Thanks,

joan,

ericromanenghi commented 8 years ago

https://github.com/andrewliao11/py-faster-rcnn/blob/master/README.md

ericromanenghi commented 8 years ago

With 2 classes is very similar, you just have to adapt the num_class and the num_output in some layers in the train.solver. If look a little more in the issues, you can find a lot of examples.