Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
968 stars 105 forks source link

how to train this model on my own datasets? #85

Open dreamlychina opened 10 months ago

dreamlychina commented 10 months ago

感谢大佬分享代码,我想问下,怎么在自己的数据集上面训练??数据集的格式跟coco一样,怎么修改类别数和类别的名称??还有一点就是,对于一张图片中既含有大目标,也含有小目标的case的话,模型的表现怎么样??

phhh-xh commented 10 months ago

You need to modify config following official instructions