Function CreateTensor

bench/BenchCvtColor.cpp:82–100

Adapted from src/util/TensorDataUtils.hpp

// Adapted from src/util/TensorDataUtils.hpp
inline static nvcv::Tensor CreateTensor(int numImages, int imgWidth, int imgHeight, const nvcv::ImageFormat &imgFormat)
{
    if (imgFormat == NVCV_IMAGE_FORMAT_NV12 || imgFormat == NVCV_IMAGE_FORMAT_NV12_ER
        || imgFormat == NVCV_IMAGE_FORMAT_NV21 || imgFormat == NVCV_IMAGE_FORMAT_NV21_ER)
    {
        if (imgHeight % 2 != 0 || imgWidth % 2 != 0)
        {
            throw std::invalid_argument("Invalid height");
        }

        int height420 = (imgHeight * 3) / 2;

        return nvcv::Tensor(numImages, {imgWidth, height420}, nvcv::ImageFormat(NVCV_IMAGE_FORMAT_Y8));
    }
    else
    {
        return nvcv::Tensor(numImages, {imgWidth, imgHeight}, imgFormat);
    }
}

template<typename BT>
inline void CvtColor(nvbench::state &state, nvbench::type_list<BT>)