Function VarShapeRemap

python/mod_cvcuda/operators/OpRemap.cpp:123–157

}

ImageBatchVarShape VarShapeRemap(ImageBatchVarShape &src, Tensor &map, NVCVInterpolationType srcInterp,
                                 NVCVInterpolationType mapInterp, NVCVRemapMapValueType mapValueType, bool alignCorners,
                                 NVCVBorderType borderMode, const pyarray &borderValue, std::optional<Stream> pstream)
{
    ImageBatchVarShape dst = ImageBatchVarShape::Create(src.capacity());

    nvcv::Size2D mapSize;

    if (mapValueType == NVCV_REMAP_ABSOLUTE || mapValueType == NVCV_REMAP_ABSOLUTE_NORMALIZED)
    {
        auto mapAccess = nvcv::TensorDataAccessStridedImagePlanar::Create(map.exportData());
        if (!mapAccess)
        {
            throw std::runtime_error("Incompatible map tensor layout");
        }

        mapSize.w = mapAccess->numCols();
        mapSize.h = mapAccess->numRows();
    }

    for (int i = 0; i < src.numImages(); ++i)
    {
        if (mapValueType == NVCV_REMAP_ABSOLUTE || mapValueType == NVCV_REMAP_ABSOLUTE_NORMALIZED)
        {
            dst.pushBack(Image::Create(mapSize, src[i].format()));
        }
        else
        {
            dst.pushBack(Image::Create(src[i].size(), src[i].format()));
        }
    }

    return VarShapeRemapInto(dst, src, map, srcInterp, mapInterp, mapValueType, alignCorners, borderMode, borderValue,
                             pstream);
}

} // namespace