From 98bb0bb24319f2cc334d4db9911373af64eb16e8 Mon Sep 17 00:00:00 2001 From: ShuangLiu1992 Date: Sun, 28 Apr 2019 04:30:03 +0100 Subject: [PATCH] Update onnx2ncnn.cpp (#948) * Update onnx2ncnn.cpp num_filter for ConvTranspose is wrong (line 1075), should be the same sa Conv, which reads the number of filters from dim(0) (line 998) int num_filter = W.dims(0); * Update onnx2ncnn.cpp --- tools/onnx/onnx2ncnn.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp index 76f362785..fdfec4b46 100644 --- a/tools/onnx/onnx2ncnn.cpp +++ b/tools/onnx/onnx2ncnn.cpp @@ -1072,7 +1072,6 @@ int main(int argc, char** argv) { const onnx::TensorProto& W = weights[node.input(1)]; - int num_filter = W.dims(1); int has_bias = node.input_size() == 3 ? 1 : 0; std::string auto_pad = get_node_attr_s(node, "auto_pad");//TODO @@ -1083,6 +1082,7 @@ int main(int argc, char** argv) std::vector output_shape = get_node_attr_ai(node, "output_shape");//TODO std::vector pads = get_node_attr_ai(node, "pads"); int group = get_node_attr_i(node, "group", 1); + int num_filter = W.dims(1) * group; fprintf(pp, " 0=%d", num_filter);