Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
yaosx425 opened this issue Jul 1, 2024 · 2 comments
Open

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

yaosx425 opened this issue Jul 1, 2024 · 2 comments

Comments

@yaosx425
Copy link

yaosx425 commented Jul 1, 2024

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

需要同时更改为num_classes=1

@yaosx425
Copy link
Author

yaosx425 commented Jul 3, 2024

需要同时更改为num_classes=1

thank you for your patient

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@yuhongtian17 @yaosx425