Merge pull request #20892 from alalek:issue_20888

This commit is contained in:
Alexander Alekhin 2021-10-19 19:45:35 +00:00
commit d33a048d89

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)