search

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
336 stars 46 forks source link

Release the code for Few-shot object detection #34

Closed 4-0-4-notfound closed 1 year ago

4-0-4-notfound commented 2 years ago

Would you like to release the code for the Airbus Ship Detection and few-shot detection part?

amirbar commented 2 years ago

I will release both of them, but it might take some more time. Mainly the FS part. For now, you can follow this Few-Shot code base (dataloaders, classes, datasplit, etc.) which is enough to reproduce the results. Edit: the FS dataset was created using seed = 0 which is one of the seeds used in that work.

4-0-4-notfound commented 2 years ago

thx~~

Flyooofly commented 1 year ago

I will release both of them, but it might take some more time. Mainly the FS part. For now, you can follow this Few-Shot code base (dataloaders, classes, datasplit, etc.) which is enough to reproduce the results. Edit: the FS dataset was created using seed = 0 which is one of the seeds used in that work.

Hello, is it convenient to open source the code of few-shot object detection part?

amirbar commented 1 year ago

Hello, apologies for dropping the ball here. Unfortunately, we've found an issue in our evaluation pipeline and you are correct. I'm now uploading a new version with a fix and will also make the few-shot checkpoint available. These are the results we are seeing:

evaluation type Novel AP 10 Novel AP 30 Novel AP75 10 Novel AP75 30
Supervised 12.4 20.4 13.3 21.8
DETReg 13.7 22.6 15.1 24.3