Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
950 stars 100 forks source link

how to use DETR aug with vit pretrain? #150

Closed egbertYeah closed 1 month ago

egbertYeah commented 1 month ago

作者你好,vit提供的config支持的LSJ数据增强,但是我的目标检测数据集没有mask,想问一下怎么关闭分割,或者如何使用DETR增强?谢谢

TempleX98 commented 1 month ago

请参考这个config的注释部分:https://github.com/Sense-X/Co-DETR/blob/d01ef8b657cc3711c59759cf863fe5ab9aa50e76/projects/configs/co_dino/co_dino_5scale_lsj_r50_1x_coco.py#L68

egbertYeah commented 1 month ago

请参考这个config的注释部分:

https://github.com/Sense-X/Co-DETR/blob/d01ef8b657cc3711c59759cf863fe5ab9aa50e76/projects/configs/co_dino/co_dino_5scale_lsj_r50_1x_coco.py#L68 好的,非常感谢