mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-11 03:33:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #20892 from alalek:issue_20888

Browse Source
This commit is contained in:
Alexander Alekhin 2021-10-19 19:45:35 +00:00
commit d33a048d89
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/int8layers/quantization_utils.cpp
View File

@ -52,9 +52,9 @@ public:
if (inputs_.depth() == CV_16S) if (inputs_.depth() == CV_16S)
{ {
UMat inputFp32(shape(inputs[0]), CV_32F); UMat inputFp32;
convertFp16(inputs[0], inputFp32); convertFp16(inputs[0], inputFp32);
inputFp32.copyTo(inputs[0]); inputs[0] = inputFp32; // replace
} }
inputs[0].convertTo(outputs[0], CV_8S, 1.f/scale, zeropoint); inputs[0].convertTo(outputs[0], CV_8S, 1.f/scale, zeropoint);
@ -118,7 +118,7 @@ public:
inputs_.getUMatVector(inputs); inputs_.getUMatVector(inputs);
outputs_.getUMatVector(outputs); outputs_.getUMatVector(outputs);
UMat outputFp32(shape(outputs[0]), CV_32F); UMat outputFp32;
inputs[0].convertTo(outputFp32, CV_32F, scale, -(scale*zeropoint)); inputs[0].convertTo(outputFp32, CV_32F, scale, -(scale*zeropoint));
if (outputs_.depth() == CV_16S) if (outputs_.depth() == CV_16S)