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 are the pretrained weights generated? How do I generate them? #6

Open
codingquark opened this issue Nov 20, 2017 · 2 comments
Open

Comments

@codingquark
Copy link

Are the pretrained weights generated by using create_yolo_caffemodel.py?

If I want to train the model on my own what would the process be? Should I first train yolo by darknet and then use create_yolo_caffemodel.py and create_yolo_prototxt.py?

@gudovskiy
Copy link
Owner

@codingquark I used pretrained weights. You can either train using darknet and then convert with the scripts or train in caffe directly.

@codingquark
Copy link
Author

Training in caffe sounds the easiest approach, since I can avoid having to convert to caffe. I have been playing with https://github.com/xingwangsfu/caffe-yolo and it is not working for many .cfgs.

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