Function GetImageVectorFromTensorPlanar

tests/common/TensorDataUtils.hpp:629–656

template<typename DT>
static void GetImageVectorFromTensorPlanar(const TensorData &tensorData, int sample, std::vector<DT> &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.");

    int elements = tDataAc->numRows() * tDataAc->numCols() * tDataAc->numChannels();

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

// sets the tensor data to the value of data, but honors strides.
template<typename DT>