fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.
Apache License 2.0
3.14k stars 513 forks source link

Is there anyone trained deformable-detr on voc2007? #189

Open order-a-lemonade opened 1 year ago

order-a-lemonade commented 1 year ago

hello, i trained deformable-detr on voc2007 recently, however the performence is not so good. my setting is: epoch=150, initLR=2e-4, lr-drop at 55epoch and 110 epoch, batchsize=1node6gpus2, the results are as follows:

image the ap50 is 71.6, and i think it's too low for deformable-detr. i had tried to use the weights which is trained on coco to evaluate on voc2007 testset, and i got 78.4 ap50.

and my loss curves are as follows: image it seems like the model overfits on train dataset very early. Is there any suggestions? or is anyone got an reasonable results on voc2007? really appreciate you guys!