amirbar / DETReg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".
https://amirbar.net/detreg
Apache License 2.0
334 stars 46 forks source link

Comparison to transfer learning based on MS COCO pre-training? #78

Open stwerner97 opened 1 year ago

stwerner97 commented 1 year ago

Hi, thanks for the great work!

Have you tried out how DETReg matches up against using a checkpoint pre-trained on e.g. MS COCO on small to medium sized datasets? The way I understand the supervised baseline is that only the backbone is pre-trained on a classification task.

Specifically, I would be interested in seeing if DETReg or DETReg + MS COCO fine-tuning can outperform transfer learning based on MS COCO.