Function SetImageTensorFromByteVectorPlanar

tests/common/TensorDataUtils.cpp:322–364

}

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

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

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

    if ((int64_t)data.size()
        != tDataAc->numCols() * tDataAc->numRows() * (tDataAc->dtype().bitsPerPixel() / 8) * tDataAc->numChannels())
        throw std::runtime_error("Data vector is incorrect size, size must be W*H*bytesPerPixel.");

    int bytesPerC = (tDataAc->dtype().bitsPerPixel() / 8);

    auto copyToGpu = [&](int j)
    {
        Byte *basePtr = tDataAc->sampleData(j);

        for (int i = 0; i < tDataAc->numChannels(); ++i)
        {
            Byte  *srcPtr        = data.data() + (i * (tDataAc->numCols() * tDataAc->numRows() * bytesPerC));
            size_t srcPitch      = tDataAc->numCols() * bytesPerC;
            size_t srcWidthBytes = tDataAc->numCols() * bytesPerC;
            if (cudaSuccess
                != cudaMemcpy2D(basePtr, tDataAc->rowStride(), srcPtr, srcPitch, srcWidthBytes, tDataAc->numRows(),
                                cudaMemcpyHostToDevice))
            {
                throw std::runtime_error("CudaMemcpy failed for channel plane copy from host to device.");
            }
            basePtr += tDataAc->planeStride();
        }
    };

    if (sample < 0)
        for (auto i = 0; i < tDataAc->numSamples(); ++i)
        {
            copyToGpu(i);
        }
    else
        copyToGpu(sample);
}

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