hgpvision / darknet

darknet深度学习框架源码分析:详细中文注释,涵盖框架原理与实现语法分析
MIT License
1.6k stars 488 forks source link

删除最后分类层的权重 #33

Open Muxindawang opened 4 years ago

Muxindawang commented 4 years ago

https://github.com/hgpvision/darknet/blob/6b90daea920e388d4ba3db7b75062a47a61fb001/src/parser.c#L1042

楼主 您好 我想问一下 我想删除掉分类模型的 后面用于分类的 卷积层的权重 是不是在这一行设置 net.n变为net.n-3就行啊(这个3是 最后的avgpool convolution softmax) 我想把darknet53.weights变成 darknet53.conv.74

Muxindawang commented 4 years ago

期待您的值点