Function TEST

tests/cvcuda/system/TestOpFindHomography.cpp:396–457

}

TEST(OpFindHomography, degenerate_identical_source_points)
{
    int numSamples = 2;
    int numPoints  = 16;

    nvcv::Tensor srcPoints(
        {
            {numSamples, numPoints},
            "NW"
    },
        nvcv::TYPE_2F32);
    nvcv::Tensor dstPoints(
        {
            {numSamples, numPoints},
            "NW"
    },
        nvcv::TYPE_2F32);
    nvcv::Tensor models(
        {
            {numSamples, 3, 3},
            "NHW"
    },
        nvcv::TYPE_F32);

    auto srcData    = srcPoints.exportData<nvcv::TensorDataStridedCuda>();
    auto dstData    = dstPoints.exportData<nvcv::TensorDataStridedCuda>();
    auto modelsData = models.exportData<nvcv::TensorDataStridedCuda>();

    std::vector<float> srcVec(2 * numSamples * numPoints);
    std::vector<float> dstVec(2 * numSamples * numPoints);
    std::vector<float> estimatedModelsVec(numSamples * 9);

    for (int i = 0; i < numSamples; i++)
    {
        float fixed_src_x = 100.0f;
        float fixed_src_y = 200.0f;

        for (int j = 0; j < numPoints; j++)
        {
            // All source points are identical
            srcVec[i * numPoints * 2 + 2 * j]     = fixed_src_x;
            srcVec[i * numPoints * 2 + 2 * j + 1] = fixed_src_y;

            // Different destination points
            dstVec[i * numPoints * 2 + 2 * j]     = j * 10.0f;
            dstVec[i * numPoints * 2 + 2 * j + 1] = j * 15.0f;
        }
    }

    ASSERT_EQ(cudaSuccess, cudaMemcpy(srcData->basePtr(), srcVec.data(), sizeof(float) * 2 * numPoints * numSamples,
                                      cudaMemcpyHostToDevice));
    ASSERT_EQ(cudaSuccess, cudaMemcpy(dstData->basePtr(), dstVec.data(), sizeof(float) * 2 * numPoints * numSamples,
                                      cudaMemcpyHostToDevice));

    cudaStream_t stream;
    ASSERT_EQ(cudaSuccess, cudaStreamCreateWithFlags(&stream, cudaStreamNonBlocking));

    cvcuda::FindHomography fh(numSamples, numPoints);