rbgirshick/rcnn

Detect objects of a single class using the PASCAL VOC 2007 pre-trained model?

BadWindshield opened this issue · 0 comments

I have a simple question -- after loading a trained model for 20 classes, can I limit the detection to a single class when I run rcnn_detect()?

I tried removing some of the strings in rcnn_model.classes but it doesn't work.