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

inconsistent description? #38

Closed d12306 closed 2 years ago

d12306 commented 2 years ago

hi, @amirbar , in paper, you describe "We pretrain two variants of DETReg based on DETR [5] and Deformable DETR [71] detectors for 5 and 60 epochs on IN1K and IN100, respectively", are you saying for both in1k and in100 datasets, we pretrain detr for 5 epochs and deformable detr for 60 epochs?

but in the model zoo table of the readme file, i see you pretrain detr for 60 epochs and deformable detr for 5 epochs, is it a typo?

amirbar commented 2 years ago

Thank you, I should definitely clarify this. Specifically, the pretraining schedules are as follows:

The fixed sentence should be: "We pretrain two variants of DETReg based on DETR [5] and Deformable DETR [71] detectors for 60 and 5 epochs on IN1K"