megvii-research / AnchorDETR

An official implementation of the Anchor DETR.
Other
338 stars 36 forks source link

关于在自己数据集上的应用 #13

Closed Breeze-Zero closed 3 years ago

Breeze-Zero commented 3 years ago

老师您好,我在尝试将我的数据应用到您的代码上,在dataset里的coco.py已改成我自己的数据格式读取,并能正常运行程序,但后续我在issues里看到别人说transformer.py里也要修改num_classes,因为代码运行里的超参并没有提到这些,担心还有地方存在疏漏,因此在这里想询问您,代码里还有哪些地方需要修改才能适配自己的数据集呢?

tangjiuqi097 commented 3 years ago

您好,如果你的类别索引小于现在设置的90的话,那么不管是否修改num_classes都是可以的,程序可以正常运行说明这个没有问题。如果你的数据量比较小的话,那最好还是finetune,可以参考这个gist