mirror of
https://github.com/opencv/opencv.git
synced 2025-06-09 18:43:05 +08:00
Merge pull request #20892 from alalek:issue_20888
This commit is contained in:
commit
d33a048d89
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user