Attempted to train on our own dataset, but an error occurred.

yuhongtian17 / Spatial-Transform-Decoupling

MIT License

89 stars 7 forks source link

Attempted to train on our own dataset, but an error occurred. #19

Open yaosx425 opened 5 months ago

yaosx425 commented 5 months ago

1719818858452 我尝试训练自己的数据集，将数据集改成dota格式，然后修改了mmrotate/mmrotate/datasets/dota.py中的classes，如图所示，但是当跑完第一个epoch进行evaluate的时候，报上面的错，我找到了报错的地方，然后对num_classes进行打印，发现仍然是打印15，所以想来问下您，还有什么地方是我没改到的吗？

yuhongtian17 commented 5 months ago

需要同时更改为num_classes=1

yaosx425 commented 5 months ago

需要同时更改为num_classes=1

thank you for your patient