提交 5c7fe0fe 编写于 作者: A Ayush Pandey 提交者: Dmitry Kurtaev

Fix Issue #13479

上级 8cedc052
...@@ -584,7 +584,7 @@ void ONNXImporter::populateNet(Net dstNet) ...@@ -584,7 +584,7 @@ void ONNXImporter::populateNet(Net dstNet)
for (int j = 1; j < node_proto.input_size(); j++) { for (int j = 1; j < node_proto.input_size(); j++) {
layerParams.blobs.push_back(getBlob(node_proto, constBlobs, j)); layerParams.blobs.push_back(getBlob(node_proto, constBlobs, j));
} }
layerParams.set("num_output", layerParams.blobs[0].size[1]); layerParams.set("num_output", layerParams.blobs[0].size[1] * layerParams.get<int>("group", 1));
layerParams.set("bias_term", node_proto.input_size() == 3); layerParams.set("bias_term", node_proto.input_size() == 3);
} }
else if (layer_type == "Transpose") else if (layer_type == "Transpose")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册