基于co_dino_5scale_lsj_vit_large_o365.pth预训练模型进行训练

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training

MIT License

1.03k stars 115 forks source link

基于co_dino_5scale_lsj_vit_large_o365.pth预训练模型进行训练 #169

Open eii-lyl opened 1 month ago

eii-lyl commented 1 month ago

您好，我的detectron数据集大小为20万。上次您发给了我这个模型：co_dino_5scale_lsj_vit_large_o365.pth。

我现在想基于co_dino_5scale_lsj_vit_large_o365.pth预训练模型进行训练，请问我把co_dino_5scale_vit_large_coco.py这个配置文件里的pretrained设置成co_dino_5scale_lsj_vit_large_o365.pth就可以了么？

TempleX98 commented 1 month ago

需要根据lvis的config进行修改