Merge pull request #20892 from alalek:issue_20888

3 years ago · d33a048d89
parent b5a9a6793b 1926e919be
commit d33a048d89
1 changed files with 3 additions and 3 deletions
--- a/modules/dnn/src/int8layers/quantization_utils.cpp
+++ b/modules/dnn/src/int8layers/quantization_utils.cpp
@ -52,9 +52,9 @@ public:

        if (inputs_.depth() == CV_16S)
        {
-            UMat inputFp32(shape(inputs[0]), CV_32F);
+            UMat inputFp32;
            convertFp16(inputs[0], inputFp32);
-            inputFp32.copyTo(inputs[0]);
+            inputs[0] = inputFp32;  // replace
        }

        inputs[0].convertTo(outputs[0], CV_8S, 1.f/scale, zeropoint);
@ -118,7 +118,7 @@ public:
        inputs_.getUMatVector(inputs);
        outputs_.getUMatVector(outputs);

-        UMat outputFp32(shape(outputs[0]), CV_32F);
+        UMat outputFp32;
        inputs[0].convertTo(outputFp32, CV_32F, scale, -(scale*zeropoint));

        if (outputs_.depth() == CV_16S)