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

How to train with my own dataset? #34

Open
knwng opened this issue Oct 19, 2017 · 1 comment
Open

How to train with my own dataset? #34

knwng opened this issue Oct 19, 2017 · 1 comment

Comments

@knwng
Copy link

knwng commented Oct 19, 2017

Hi, thanks for sharing this excellent project. I've successfully trained it on VOC dataset and got pretty good result and speed, but when I try to train it using my own dataset(with only 1 class). It always get 0 mAP and nan loss as testing.
I've already change the num_class in detect_loss layer and associated params in reg_reshape and the last conv layer and I've pre-proccessed my input data(reshape and subtract mean value).
What confuse me is the meaning of the other params in detect_loss layer like coord_scale, noobject_scale...Should I change them for new dataset? If so, how should I calculate them? Thanks in advance!

@Harick1
Copy link
Owner

Harick1 commented Nov 30, 2017

@knwng I use the default parameters in the original paper. To change the hyper parameters and adapt it to your own dataset, please refer to the original paper.

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

2 participants