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

num_class #11

Open
linchunmian opened this issue Sep 28, 2020 · 1 comment
Open

num_class #11

linchunmian opened this issue Sep 28, 2020 · 1 comment

Comments

@linchunmian
Copy link

你好,请问下如果我直接用DETRAC数据集来训练模型的话,num_class是设置为1还是4呢?我看数据集官网是细分了4中车辆类别,不过我最后只想把训练好的模型用来做车辆跟踪,不需要区分具体的车型,这种情况的话是不是num_class=1就可以了呢?
另一方面我是考虑如果直接设置为1是不是会影响模型训练的效果呢?我看label文件中类别id是从0到3的,所以能否给个建议?
感谢!

@linchunmian
Copy link
Author

还有个问题也想请问下,下图:
Screenshot from 2020-09-28 23-19-58
Screenshot from 2020-09-28 23-19-43
我是使用DETRAC的cfg文件作为data_cfg,但是这边的data_dir指的是什么呢,我不太明白这地方是否需要修改;此外在下方的dataset部分这里用的是jde,是否需要修改为其他呢?

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