Function AdvCvtColor

python/mod_cvcuda/operators/OpAdvCvtColor.cpp:47–55

}

Tensor AdvCvtColor(Tensor &input, NVCVColorConversionCode code, NVCVColorSpec spec, std::optional<Stream> pstream)
{
    nvcv::ImageFormat outputFormat = GetOutputFormat(input.dtype(), code);
    nvcv::TensorShape outputShape  = GetOutputTensorShape(input.shape(), outputFormat, code);

    Tensor output = Tensor::Create(outputShape, input.dtype());

    return AdvCvtColorInto(output, input, code, spec, pstream);
}

} // namespace