detector as feature extractor

[gaussiangit]

Hello,

I am using mask rcnn for detection. So basically fine tuning. However I want extract feature for each object that is being detected.
So possibly extracting feature vector just before last layer. How can I do that ? forward hooks ?

I was also looking into https://github.com/pytorch/vision/blob/master/torchvision/models/_utils.py ? could not get it working.

Also how to use jit for the same ?

Any leads would be helpful. @fmassa

Cheers!