Function GetImageByteVectorFromTensorPlanar

tests/common/TensorDataUtils.cpp:262–292

}

static void GetImageByteVectorFromTensorPlanar(const TensorData &tensorData, int sample,
                                               std::vector<nvcv::Byte> &outData)
{
    Optional<TensorDataAccessStridedImagePlanar> tDataAc = nvcv::TensorDataAccessStridedImagePlanar::Create(tensorData);

    if (!tDataAc)
        throw std::runtime_error("Tensor Data not compatible with planar access.");

    if (tDataAc->numSamples() <= sample || sample < 0)
        throw std::runtime_error("Number of samples smaller than requested sample.");

    // in a planar tensor the dtype represents each plane so the total bytes per pixel must be calculated
    int bytesPerC       = tDataAc->dtype().bitsPerPixel() / 8;
    int outputSizeBytes = tDataAc->numRows() * tDataAc->numCols() * bytesPerC * tDataAc->numChannels();

    // Make sure we have the right size.
    outData.resize(outputSizeBytes);
    Byte  *basePtr  = tDataAc->sampleData(sample);
    size_t dstWidth = tDataAc->numCols() * bytesPerC;
    for (int i = 0; i < tDataAc->numChannels(); ++i)
    {
        if (cudaSuccess
            != cudaMemcpy2D(outData.data() + (i * (tDataAc->numCols() * tDataAc->numRows()) * bytesPerC), dstWidth,
                            basePtr, tDataAc->rowStride(), dstWidth, tDataAc->numRows(), cudaMemcpyDeviceToHost))
        {
            throw std::runtime_error("CudaMemcpy failed on copy of channel plane from device to host.");
        }
        basePtr += tDataAc->planeStride();
    }
    return;
}

void GetImageByteVectorFromTensor(const TensorData &tensorData, int sample, std::vector<nvcv::Byte> &outData)
{