@fmassa I have adopted code from https://github.com/pytorch/vision/blob/master/torchvision/models/detection/faster_rcnn.py 4 and implemented
https://github.com/nithinraok/VisualQuestion_VQA/blob/master/common_resources/rpn_test_2.ipynb 3 to get RPN output. But unfortunately I am unable to get Kx2048 vector output. Instead I am getting 14x4. Can you please help me with this?