Function FillTensorData

src/nvcv/src/priv/TensorData.cpp:50–189

}

void FillTensorData(IImage &img, NVCVTensorData &tensorData)
{
    ImageFormat fmt = img.format();

    // Must do a lot of checks to see if image is compatible with a tensor representation.

    if (img.format().memLayout() != NVCV_MEM_LAYOUT_PL)
    {
        throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "Image format's memory layout must be pitch-linear";
    }

    if (img.format().css() != NVCV_CSS_444)
    {
        throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "Image format's memory layout must not have sub-sampled planes";
    }

    for (int p = 1; p < fmt.numPlanes(); ++p)
    {
        if (fmt.planeDataType(p) != fmt.planeDataType(0))
        {
            throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "Data type of all image planes must be the same";
        }
    }

    NVCVImageData imgData;
    img.exportData(imgData);

    if (imgData.bufferType != NVCV_IMAGE_BUFFER_STRIDED_CUDA)
    {
        throw Exception(NVCV_ERROR_INVALID_ARGUMENT)
            << "Only cuda-accessible images with pitch-linear data are accepted";
    }

    NVCVImageBufferStrided &imgStrided = imgData.buffer.strided;

    for (int p = 1; p < imgStrided.numPlanes; ++p)
    {
        if (imgStrided.planes[p].width != imgStrided.planes[0].width
            || imgStrided.planes[p].height != imgStrided.planes[0].height)
        {
            throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "All image planes must have the same dimensions";
        }

        if (imgStrided.planes[p].rowStride != imgStrided.planes[0].rowStride)
        {
            throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "All image planes must have the same row pitch";
        }

        if (imgStrided.planes[p].basePtr <= imgStrided.planes[0].basePtr)
        {
            throw Exception(NVCV_ERROR_INVALID_ARGUMENT)
                << "Consecutive image planes must have increasing memory addresses";
        }

        if (p >= 2)
        {
            intptr_t planeStride = reinterpret_cast<const std::byte *>(imgStrided.planes[1].basePtr)
                                 - reinterpret_cast<const std::byte *>(imgStrided.planes[0].basePtr);