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

I detect nothing when using my own dataset to train. #96

Open
ZoeSnow opened this issue May 20, 2019 · 7 comments
Open

I detect nothing when using my own dataset to train. #96

ZoeSnow opened this issue May 20, 2019 · 7 comments

Comments

@ZoeSnow
Copy link

ZoeSnow commented May 20, 2019

I trained on my own dataset and replaced the yolo_small.ckpt with data,meta and index files, but no objects could be detected.(好桑心) Does anyone know what I should do?Thank you!!!

@Yusei37
Copy link

Yusei37 commented May 27, 2019

I trained on my own dataset and replaced the yolo_small.ckpt with data,meta and index files, but no objects could be detected.(好桑心) Does anyone know what I should do?Thank you!!!

I have the same problem, too. Did you solve it?
I trained it without yolo_small.ckpt, and the code run normally.

@ZoeSnow
Copy link
Author

ZoeSnow commented May 28, 2019

I haven't solved the error yet. My training process is also normal but I can't detect anything.

@zuowk
Copy link

zuowk commented May 29, 2019

the loss layer in the training file seems to be wrong,so training on your your dataset can't work

@ZoeSnow
Copy link
Author

ZoeSnow commented May 30, 2019

@zuowk

Thank you for your reply. Did you train on your own dataset successfully? Could you tell me how to rectify the loss function?

@JokerKangS
Copy link

the loss layer in the training file seems to be wrong,so training on your your dataset can't work

I trained it on the pascal_voc dataset without yolo_small.ckpt, and still can't detect anything, too.

@Lizhihzzx
Copy link

I have the same problem.Did anybody solve it?Or it means that i must use yolo_small.ckpt?

@JokerKangS
Copy link

I found the reason!when i review the origin paper of yolo,i found that the auther pretrain the first 20 convonlutional layers on the ImageNet 1000-class dataset. So i pretrain the first conv-layers on another classification dataset(because i have no imagenet dateset),then train the whole network. Finally, it can detect things correctly! In short, if you want to train yolo from scratch, you should pretrain the first 20 conv-layers with a classification task

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

5 participants