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

Did you fine-tune a pre-trained YOLOv3 or train this from scratch? #1

Open
Karthik-Suresh93 opened this issue Apr 19, 2018 · 9 comments

Comments

@Karthik-Suresh93
Copy link

As you have only a few images here, I am curious if you trained on this dataset from scratch or fine-tuned a few layers in the darknet architecture using a pre-trained model? Could you please share your training strategy?

@jekhor
Copy link
Owner

jekhor commented Apr 19, 2018 via email

@Karthik-Suresh93
Copy link
Author

Karthik-Suresh93 commented Apr 19, 2018

But you have very few images to train on, it didn't produce any results for me. Thanks for your help anyway

@jekhor
Copy link
Owner

jekhor commented Apr 19, 2018

This dataset contains ~4000 labeled objects, this is not too small amount. But the net trained with this data doesn't show good results with photos from completely different series (taken with different device, weather conditions, zoom etc.) and you need additional samples in such case. I can recommend to take a look for openaerialmap.org for example (select filtering by high resolution).

@jekhor
Copy link
Owner

jekhor commented Apr 19, 2018

This is result from my current net applied to the San Francisco's photo from openaerialmap, for example.

predictions

@Karthik-Suresh93
Copy link
Author

Could you provide a link to the full dataset, please? Your uploads contain about 130 images

@jekhor
Copy link
Owner

jekhor commented Apr 19, 2018 via email

@Karthik-Suresh93
Copy link
Author

Did you freeze any weights or train the full model?

@jekhor
Copy link
Owner

jekhor commented Apr 21, 2018

Full train as described here: https://pjreddie.com/darknet/yolo/
This was my first attempt to get some results with YOLO, so I just repeated instructions written by its author.

@Karthik-Suresh93
Copy link
Author

Thank you very much for your help!

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