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

Train new pretrained model #278

Open
StefanCola opened this issue May 5, 2018 · 5 comments
Open

Train new pretrained model #278

StefanCola opened this issue May 5, 2018 · 5 comments

Comments

@StefanCola
Copy link

Hello,

All the pretrained models didn't result in good accuracy.

So I decided to train my own pretrained model.

I have very small grasses to detect, in an image of 1920x1080 they aren't bigger than 100x100 pixels. So I trained Full Yolo on a 100x100 dataset, my accuracy is roundabout 90%.

When I use these model weights as Backend_pretrained model and start training yolo on HD images, the accuracy starts at 0.15 and degrees down to zero. I have no Recalls.

I can't find my mistake. Isn't yolo able to detect object with large x and very small y dimension? Do I have to note something by training the pretrained model?

@rodrigo2019
Copy link

I got a similar problem, my dataset has resolution 640x480 and my objects has around 80x30, to increase my performance I removed a convolution block from TinyDarknet, so the grid got a size of 16x16 instead 8x8

@ZacharyForrest
Copy link

Presumably you'll need to adjust your anchors when training on the HD images - are you doing that?

@StefanCola
Copy link
Author

StefanCola commented May 8, 2018

I am generating the anchors as shown in the readme file with gen_anchors. I am also confused because the anchor ratios are near to zero

@shivaram93
Copy link

how do we generate own anchors for our custom dataset with the help of gen_anchors.py?

@mdawas
Copy link

mdawas commented Oct 22, 2018

@shivaram93 you just need to run the script and pass the config file as an argument. Make sure to adjust the config file to match your needs.

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