How could i do if i want to train the semi-detr on a custom dataset ?

JCZ404 / Semi-DETR

[CVPR 2023] Official implementation of the paper "Semi-DETR: Semi-Supervised Object Detection with Detection Transformers"

https://arxiv.org/abs/2307.08095

MIT License

77 stars 9 forks source link

How could i do if i want to train the semi-detr on a custom dataset ? #19

Closed meftehs closed 5 months ago

JCZ404 commented 5 months ago

Hi, I think the minimum modification is to prepare your dataset in the COCO format, and then utilize the implemented dataset in this repo to load your dataset.