Obtain the objectness heat maps for a given image by Faster-RCNN from torchvision
Obtain the objectness heat maps for a given image by Faster-RCNN from torchvision
Inspired by this blog @hirotoschwert/digging-into-detectron-2-part-4-3d1436f91266">Digging into Detectron 2, the heat maps from the part 4 are interesting to reproduce. Though that blog used Detectron2, this repo reproduces the same results with torchvision only.
Just refer to Demo_Objectness.ipynb
, or run a Colab version here: