Closed Breeze-Zero closed 3 years ago
老师您好,我在尝试将我的数据应用到您的代码上,在dataset里的coco.py已改成我自己的数据格式读取,并能正常运行程序,但后续我在issues里看到别人说transformer.py里也要修改num_classes,因为代码运行里的超参并没有提到这些,担心还有地方存在疏漏,因此在这里想询问您,代码里还有哪些地方需要修改才能适配自己的数据集呢?
您好,如果你的类别索引小于现在设置的90的话,那么不管是否修改num_classes都是可以的,程序可以正常运行说明这个没有问题。如果你的数据量比较小的话,那最好还是finetune,可以参考这个gist 。
num_classes
老师您好,我在尝试将我的数据应用到您的代码上,在dataset里的coco.py已改成我自己的数据格式读取,并能正常运行程序,但后续我在issues里看到别人说transformer.py里也要修改num_classes,因为代码运行里的超参并没有提到这些,担心还有地方存在疏漏,因此在这里想询问您,代码里还有哪些地方需要修改才能适配自己的数据集呢?