tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 399 forks source link

Getting RPN's output after training Faster R-CNN model with use_rcnn=True #282

Open munikarmanish opened 5 years ago

munikarmanish commented 5 years ago

I have trained a Faster R-CNN model (including the final RCNN module). Now the final output contains bounding boxes and class probabilities. But I want to know the class-independent bounding boxes and their objectness scores (which is what the RPN module gives) from the same model. Can anyone help me how to get them in Luminoth?

Any help is much appreciated!