-
Notifications
You must be signed in to change notification settings - Fork 2.8k
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
Darknet cfg convolution weights dimensions #370
Comments
Hi, Looking at ONNX which is the most mature in this context, they still have major problems with their shape inference. If you can think of other simpler ways to fix these kinds of issues, I may be willing to help in the near future. I didn't try but I suspect other neural network formats that are missing intermediate tensor shape info can have this issue as well. Cheers |
I noticed that when I use Darknet's .cfg files and visualize the network topology, I got 4D weights like 3x3x3x(# of filters). I think you are using 'channels' information in the .cfg file for all weights channels. This seems to be wrong since the input feature map channel dimension is equal to the previous convolution # of filters. Also, see the actual page for detailed rundown of this topology.
Tiny Darknet
If I missed something or have a false assumption, feel free to correct me ;)
The text was updated successfully, but these errors were encountered: