../lib/roi_data_layer/layer.py", line 125, in setup top[idx].reshape(1, self._num_classes * 4) IndexError: Index out of range

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

../lib/roi_data_layer/layer.py", line 125, in setup top[idx].reshape(1, self._num_classes * 4) IndexError: Index out of range #854

Open zhoupan9109 opened 6 years ago

zhoupan9109 commented 6 years ago

I train faster-rcnn based on resnet50 and change the classes number to 5, when I run the demo.py, the error run out! How can I solve this problem? thank you very much!

zhoupan9109 commented 6 years ago

the --weights net's classes number is 20