I would like to extract the features after ROI Align in object detection.
Specifically for every box detected (top k boxes), I would like to extract the backbone features, either the FC layer before classification or the layer before that in the backbone.
Can anyone suggest a straightforward way of doing this or provide a short script?
Thanks