diff --git a/modules/dnn/src/onnx/onnx_importer.cpp b/modules/dnn/src/onnx/onnx_importer.cpp index 423bde5474..a548975642 100644 --- a/modules/dnn/src/onnx/onnx_importer.cpp +++ b/modules/dnn/src/onnx/onnx_importer.cpp @@ -1168,6 +1168,7 @@ void ONNXImporter::parseGlobalPool(LayerParams &layerParams, const opencv_onnx:: void ONNXImporter::parseReduce(LayerParams& layerParams, const opencv_onnx::NodeProto& node_proto) { + layerParams.type = "Reduce"; const auto& op_type = node_proto.op_type(); String reduce_type; if (op_type == "ReduceMax") @@ -1209,9 +1210,16 @@ void ONNXImporter::parseReduce(LayerParams& layerParams, const opencv_onnx::Node for (int i = 0; i < num_axes; ++i) axes[i] = mat_axes.at(i); layerParams.set("axes", DictValue::arrayInt(&axes[0], num_axes)); + if (constBlobs.find(node_proto.input(0)) != constBlobs.end()){ + std::vector inputs, output; + inputs.push_back(getBlob(node_proto, 0)); + runLayer(layerParams, inputs, output); + CV_Assert(output.size() == 1); + addConstant(node_proto.output(0), output[0]); + return; + } } - layerParams.type = "Reduce"; addLayer(layerParams, node_proto); } diff --git a/modules/dnn/test/test_onnx_importer.cpp b/modules/dnn/test/test_onnx_importer.cpp index f6d05464eb..743c19a5e6 100644 --- a/modules/dnn/test/test_onnx_importer.cpp +++ b/modules/dnn/test/test_onnx_importer.cpp @@ -1550,6 +1550,10 @@ TEST_P(Test_ONNX_layers, Einsum_const_inputs) { testONNXModels("einsum_const_inputs", npy, 0, 0, false, false, 1); } +TEST_P(Test_ONNX_layers, ReduceSum_Consts){ + testONNXModels("reducesum_consts"); +} + TEST_P(Test_ONNX_layers, Pad2d_Unfused) { testONNXModels("ReflectionPad2d");