-
Notifications
You must be signed in to change notification settings - Fork 47
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
Running command #4
Comments
@joohaeng - yes the '-w' argument is for specifying what weights you want to load into the model. By default, the model will load the pre-trained weights I included in the 'weights' folder. You only need to specify the weights argument if you want to load your own pre-trained weights. Example using weights included in repo:
|
@BelalC Thanks! By the way, which pre-trained net shows the best performance? |
VGG16 was the most accurate the last time I tested it out, but it's fairly compute heavy. MobileNet was the fastest but least accurate, although it could definitely be improved with some more tweaking + data. Let me know how you go! |
@BelalC, When running the demo, I specify the name of the pre-trained net using "-m" option. But, I could not make it work using "-weights" option. Is it related to the files in "weights" folder? Could you give some working example?
The text was updated successfully, but these errors were encountered: