training with custom data set

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training

MIT License

968 stars 105 forks source link

training with custom data set #67

Closed byungjinku closed 4 months ago

byungjinku commented 11 months ago

Hello can we use your code with my custom data set?

Best regard,

TempleX98 commented 11 months ago

Sure, you can train Co-DETR on your own custom data with the pre-trained weights. For more details about custom data training, please refer to the official tutorial