I am using the pytorch Faster R-CNN model with a ResNet-50-FPN backbone. I used this tutorial https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html 1
The output at the “TwoMLPHeads” comes out to be (512,1024)