Hi, I want to extend the torchvision’s Faster RCNN
(vision/faster_rcnn.py at main · pytorch/vision · GitHub). My plan is to have an additional classification task where features from the
feature extractor are used to determine the class of the image. Then the classification loss will be computed and backpropagated to the
feature extractor. How do you go about it?
Apologies for a novice question.