Co-DETR 使用自己的训练数据集训练实例分割任务不收敛

使用自己的训练数据集，转换成coco2017格式，20个类别，checkpoint使用

./tools/dist_train.sh /workspace/Co-DETR/projects/configs/co_dino_vit/co_dino_5scale_vit_large_coco_instance.py 4 /workspace/train_workdir 训练100 epoch，loss_mask: 20.5192，推理结果完全不对，除了数据集相关类别标签、数量之外，还有哪里需要修改吗？

Apr 27 '25 07:04 gujiacheng

使用自己的训练数据集，转换成coco2017格式，20个类别，checkpoint使用

./tools/dist_train.sh /workspace/Co-DETR/projects/configs/co_dino_vit/co_dino_5scale_vit_large_coco_instance.py 4 /workspace/train_workdir 训练100 epoch，loss_mask: 20.5192，推理结果完全不对，除了数据集相关类别标签、数量之外，还有哪里需要修改吗？

你好，我能先问问你修改了什么才能进行实例分割么

Apr 28 '25 01:04 DaBaiTuu

使用自己的训练数据集，转换成coco2017格式，20个类别，checkpoint使用 ./tools/dist_train.sh /workspace/Co-DETR/projects/configs/co_dino_vit/co_dino_5scale_vit_large_coco_instance.py 4 /workspace/train_workdir 训练100 epoch，loss_mask: 20.5192，推理结果完全不对，除了数据集相关类别标签、数量之外，还有哪里需要修改吗？

你好，我能先问问你修改了什么才能进行实例分割么

使用对应的配置文件就可以

Apr 28 '25 03:04 gujiacheng

请问你的实例分割训练自己数据集收敛了吗，我这边训也是不收敛

Aug 08 '25 03:08 dingliangxiansheng