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.

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

训练自己的数据集除了修改配置文件还需要修改mmdet中的文件么? #6

Open
YingjianYu opened this issue Dec 11, 2023 · 0 comments

Comments

@YingjianYu
Copy link

将自己的关键点检测数据集转为coco格式后,只修改配置文件开始训练。训练能够跑起来,但是loss一直为0,按照网上的教程进行了如下修改:
第一处: mmdet/core/evalution/class_names.py 代码下的 def coco_classes() 的 return 内容改为自己数据集的类别;
第二处:mmdet/datasets/coco.py 代码下的 class CocoDataset(CustomDataset) 的 CLASSES 改为自己数据集的类别。
程序才跑起来,请问自豪兄是否有只修改配置文件的方法,不想动mmdetection里的文件。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant