Merge pull request #3648 from asmorkalov:al/dnn-type-inference

Added dnn type inference support
1 year ago · d4a1aeaff4
parent cf63a7f71f dd7d7e98a2
commit d4a1aeaff4
1 changed files with 1 additions and 1 deletions
--- a/modules/text/src/ocr_holistic.cpp
+++ b/modules/text/src/ocr_holistic.cpp
@ -75,7 +75,7 @@ protected:
        inputShape.push_back(getPerceptiveField().height);
        inputShape.push_back(getPerceptiveField().width);
        vector<dnn::MatShape> inShapes, outShapes;
-        net.getLayerShapes(inputShape, id, inShapes, outShapes);
+        net.getLayerShapes(inputShape, CV_32F, id, inShapes, outShapes);
        CV_Assert(outShapes.size() == 1 && outShapes[0].size() == 4);
        CV_Assert(outShapes[0][0] == 1 && outShapes[0][2] == 1 && outShapes[0][3] == 1);
        return outShapes[0][1];