diff --git a/common/waifu2x.cpp b/common/waifu2x.cpp index e57f047..7e3ff5e 100644 --- a/common/waifu2x.cpp +++ b/common/waifu2x.cpp @@ -1652,7 +1652,7 @@ Waifu2x::eWaifu2xError Waifu2x::AfterReconstructFloatMatProcess(const bool isRec const double ratio = CalcScaleRatio(image_size); const int scale2 = ceil(log2(ratio)); - const double shrinkRatio = ratio / std::pow(2.0, (double)scale2); + const double shrinkRatio = ratio >= 1.0 ? ratio / std::pow(2.0, (double)scale2) : ratio; cv::Mat alpha; if (floatim.channels() == 4)