search

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.15k stars 232 forks source link

Training with custom dataset #134

Open a-sekhar opened 1 year ago

a-sekhar commented 1 year ago

Hi, Thank you for your beautiful work. I am trying to train using custom dataset. I have tried replicating the COCODIR format for the training but unable to do so. Could you kindly provide a manual to work with custom dataset, and how custom dataset format should look like? Thank you.

SlongLiu commented 1 year ago

We provide a guide on custom dataset in https://github.com/IDEA-Research/DINO#run. Training/Fine-tuning a DINO on your custom dataset item. Hope this section is helpful.