open-mmlab / mmdetection

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

Models for Self-Supervised Downstream Detection Tasks #10831

Open chos1npc opened 1 year ago

chos1npc commented 1 year ago

I would like to ask whether it is possible to train the model transferred from self-supervision through mmdet. Is the naming of the backbone consistent with most self-supervised models?

hhaAndroid commented 1 year ago

@chos1npc Self-supervised training is currently done using mmpretrain, and the resulting models can be directly used in MMDetection. You can perform self-supervised training using the mmpretrain library and then directly utilize the trained models in the MMDetection framework for tasks such as object detection.

https://github.com/open-mmlab/mmpretrain/blob/main/docs/en/user_guides/downstream.md

chos1npc commented 1 year ago

@hhaAndroid Is it possible to transfer a self-supervised model that wasn't trained using mmpretrain?

hhaAndroid commented 1 year ago

@chos1npc Of course, you can simply transfer some keys or directly copy the backbone code and register it to achieve that.