Merge pull request #13882 from dkurt:fix_13479
This commit is contained in:
@@ -1217,12 +1217,8 @@ public:
|
||||
int dims[] = {inputs[0][0], outCn, outH, outW};
|
||||
outputs.resize(inputs.size(), shape(dims, 4));
|
||||
|
||||
internals.push_back(MatShape());
|
||||
if (!is1x1())
|
||||
internals[0] = computeColRowShape(inputs[0], outputs[0]);
|
||||
|
||||
if (hasBias())
|
||||
internals.push_back(shape(1, outH*outW));
|
||||
internals.push_back(computeColRowShape(inputs[0], outputs[0]));
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
@@ -584,7 +584,7 @@ void ONNXImporter::populateNet(Net dstNet)
|
||||
for (int j = 1; j < node_proto.input_size(); 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);
|
||||
}
|
||||
else if (layer_type == "Transpose")
|
||||
|
||||
@@ -72,6 +72,7 @@ TEST_P(Test_ONNX_layers, Deconvolution)
|
||||
{
|
||||
testONNXModels("deconvolution");
|
||||
testONNXModels("two_deconvolution");
|
||||
testONNXModels("deconvolution_group");
|
||||
}
|
||||
|
||||
TEST_P(Test_ONNX_layers, Dropout)
|
||||
|
||||
Reference in New Issue
Block a user