Function PillowResizeInto

python/mod_cvcuda/operators/OpPillowResize.cpp:178–204

};

Tensor PillowResizeInto(Tensor &output, Tensor &input, nvcv::ImageFormat format, NVCVInterpolationType interp,
                        std::optional<Stream> pstream)
{
    if (!pstream)
    {
        pstream = Stream::Current();
    }
    auto in_access  = nvcv::TensorDataAccessStridedImagePlanar::Create(input.exportData());
    auto out_access = nvcv::TensorDataAccessStridedImagePlanar::Create(output.exportData());
    if (!in_access || !out_access)
    {
        throw std::runtime_error("Incompatible input/output tensor layout");
    }

    // Use CreateOperatorEx to use the extended create operator function passing the specialized PyOperator above
    // as template type, instead of the regular cvcuda::OP class used in the CreateOperator function.
    auto pillowResize = CreateOperatorEx<PyOpPillowResize>();

    ResourceGuard guard(*pstream);
    guard.add(LockMode::LOCK_MODE_READ, {input});
    guard.add(LockMode::LOCK_MODE_WRITE, {output});
    guard.add(LockMode::LOCK_MODE_READWRITE, {*pillowResize});

    pillowResize->submit(pstream->cudaHandle(), input, output, format, interp);

    return output;
}

Tensor PillowResize(Tensor &input, const Shape &out_shape, nvcv::ImageFormat format, NVCVInterpolationType interp,
                    std::optional<Stream> pstream)