Function FindHomography

bench/BenchFindHomography.cpp:79–130

template<typename T>
inline void FindHomography(nvbench::state &state, nvbench::type_list<T>)
try
{
    long2 shape      = benchutils::GetShape<2>(state.get_string("shape"));
    bool  batch      = state.get_int64("batch");
    auto  numSamples = batch ? 1 : shape.x;

    nvcv::Tensor src({{numSamples, shape.y}, "NW"}, nvcv::TYPE_2F32);
    nvcv::Tensor dst({{numSamples, shape.y}, "NW"}, nvcv::TYPE_2F32);
    nvcv::Tensor models({{numSamples, 3, 3}, "NHW"}, benchutils::GetDataType<T>());

    std::vector<T> srcVec(2 * numSamples * shape.y);
    std::vector<T> dstVec(2 * numSamples * shape.y);
    std::vector<T> modelsVec(9);

    fill_vector(srcVec);
    fill_vector(modelsVec);
    fill_dst(srcVec, modelsVec, dstVec, numSamples);
    fill_tensor(src, srcVec);
    fill_tensor(dst, dstVec);

    state.add_global_memory_reads(shape.x * shape.y * 4 * sizeof(T));
    state.add_global_memory_writes(shape.x * 3 * 3 * sizeof(T));

    cvcuda::FindHomography op(shape.x, shape.y);

    if (!batch) // negative var shape means use Tensor
    {
        state.exec(nvbench::exec_tag::sync, [&op, &src, &dst, &models](nvbench::launch &launch)
        {
            op(launch.get_stream(), src, dst, models);
        });
    }
    else
    {
        nvcv::TensorBatch srcTensors(shape.x);
        nvcv::TensorBatch dstTensors(shape.x);
        nvcv::TensorBatch modelsTensors(shape.x);
        srcTensors.pushBack(src);
        dstTensors.pushBack(dst);
        modelsTensors.pushBack(models);

        state.exec(nvbench::exec_tag::sync, [&op, &srcTensors, &dstTensors, &modelsTensors](nvbench::launch &launch)
        {
            op(launch.get_stream(), srcTensors, dstTensors, modelsTensors);
        });
    }
}
catch (const std::exception &err)
{
    state.skip(err.what());
}

// clang-format on