search

open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
29.01k stars 9.36k forks source link

During object detection, How to extract image feature from neck layer during eval mode in MMDET 3.3.0? #11825

Open prafful-kumar opened 2 months ago

prafful-kumar commented 2 months ago

For example, I have a faster-rcnn model pre-trained on coco dataset and I am fine-tuning on downstream dataset. To analyse the features of the detector, I want to extract the embeddings from the neck (i.e., FPN), but I am not sure how to extract the feature embeddings. Can you help me understand and implement it?

PeterVennerstrom commented 1 month ago

The detector level abstraction is one place you could access the embeddings.