Function Normalize

tests/cvcuda/system/TestOpNormalize.cpp:34–79

namespace t    = ::testing;

static void Normalize(std::vector<uint8_t> &hDst, int dstRowStride, const std::vector<uint8_t> &hSrc, int srcRowStride,
                      nvcv::Size2D size, nvcv::ImageFormat fmt, const std::vector<float> &hBase, int baseRowStride,
                      nvcv::Size2D baseSize, nvcv::ImageFormat baseFormat, const std::vector<float> &hScale,
                      int scaleRowStride, nvcv::Size2D scaleSize, nvcv::ImageFormat scaleFormat,
                      const float globalScale, const float globalShift, const float epsilon, const uint32_t flags)
{
    using FT = float;

    for (int i = 0; i < size.h; i++)
    {
        const int bi = baseSize.h == 1 ? 0 : i;
        const int si = scaleSize.h == 1 ? 0 : i;

        for (int j = 0; j < size.w; j++)
        {
            const int bj = baseSize.w == 1 ? 0 : j;
            const int sj = scaleSize.w == 1 ? 0 : j;

            for (int k = 0; k < fmt.numChannels(); k++)
            {
                const int bk = (baseFormat.numChannels() == 1 ? 0 : k);
                const int sk = (scaleFormat.numChannels() == 1 ? 0 : k);

                FT mul;

                if (flags & CVCUDA_NORMALIZE_SCALE_IS_STDDEV)
                {
                    FT s = hScale.at(si * scaleRowStride + sj * scaleFormat.numChannels() + sk);
                    FT x = s * s + epsilon;
                    mul  = FT{1} / std::sqrt(x);
                }
                else
                {
                    mul = hScale.at(si * scaleRowStride + sj * scaleFormat.numChannels() + sk);
                }

                FT res = std::rint((hSrc.at(i * srcRowStride + j * fmt.numChannels() + k)
                                    - hBase.at(bi * baseRowStride + bj * baseFormat.numChannels() + bk))
                                       * mul * globalScale
                                   + globalShift);

                hDst.at(i * dstRowStride + j * fmt.numChannels() + k) = res < 0 ? 0 : (res > 255 ? 255 : res);
            }
        }
    }
}

static uint32_t normalScale   = 0;
static uint32_t scaleIsStdDev = CVCUDA_NORMALIZE_SCALE_IS_STDDEV;