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

utils.load_weights doesn't work for custom number of classes #449

Open
dinesh-tadi opened this issue Jun 8, 2022 · 1 comment
Open

utils.load_weights doesn't work for custom number of classes #449

dinesh-tadi opened this issue Jun 8, 2022 · 1 comment

Comments

@dinesh-tadi
Copy link

Loading yolov4.weights for 80 classes works fine but I wanted to use yolov4.conv.137 for the custom number of classes (as this is available in the official darknet repo for training for the custom number of classes, I read that this weights file contains the weights for the layers except for the Yolo layer), but it doesn't work and gives all outputs as nan.
Any suggestions on how to modify the utils.load_weights function so that I can load this file?

@ahoereth
Copy link

Related: #180 & #181

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