Train a Faster-RCNN in PyTorch

How can i train Faster RCNN with a different backbone other than ResNet-50 (as implemented in this tutorial https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html) ?

Let’s say that i already have a trained CNN that can classify cats and dogs how can i make this a backbone for the Faster RCNN network to detect cats and dogs.

  • How to construct this network ?
  • How do i train it?
  • Also iI would like to implement my own training loop rather the using the one provided by the PyTorch team ? What would a general training loop for this model look like ?