search

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
1.02k stars 113 forks source link

Self-dataset training problem #39

Closed shuaixinxin closed 1 year ago

shuaixinxin commented 1 year ago

problem: AssertionError: The num_classes (4) in CoDeformDETRHead of MMDataParallel does not matches the length of CLASSES 80) in CocoDataset

shuaixinxin commented 1 year ago

作者您好,我自己的数据集只有4个类别应该如何修改呢？我想修改训练时的图像尺寸的大小，应该如何修改代码呢？望指导，万分感谢！

TempleX98 commented 1 year ago

@shuaixinxin , Hi, you can refer to this tutorial for custom dataset training. The image resolution can be modified in the config.