fredzzhang/upt

How about training a DETR model on the VCOCO dataset

Closed this issue · 4 comments

Thank you for your excellent work.
you have provided a tutorial on training DETR models on the HICO-DET dataset, could you tell us how you trained the DETR on the VCOCO dataset?

Hi @ddwhzh,

The code for fine-tuning on V-COCO can be found in the same repository, under the branch vcoco, you do need to set up a softlink pointing to the V-COCO dataset though.

Cheers,
Fred.

Sorry for the possible misunderstanding due to the expression, I would like to know how to pre-train a detr pre-train model on vcoco instead of training upt with a pre-trained model.

The vcoco branch is precisely for that purpose. If you look at the history of the branch, you'll find that commit 906fb89 added support for fine-tuning the object detector on V-COCO.

THX, understood.