Function TEST

tests/cvcuda/system/TestOpMinAreaRect.cpp:94–171

}

TEST(OpMinAreaRect, MinAreaRect_sanity)
{
    int batchsize = 3;

    std::vector<std::vector<short>> contourPointsData;

    contourPointsData.push_back(
        {845, 600, 845, 601, 847, 603, 859, 603, 860, 604, 865, 604, 866, 603, 867, 603, 868, 602, 868, 601, 867, 600});
    contourPointsData.push_back({965,  489, 964,  490, 963,  490, 962,  491, 962,  494, 963,  495,
                                 963,  499, 964,  500, 964,  501, 966,  503, 1011, 503, 1012, 504,
                                 1013, 503, 1027, 503, 1027, 502, 1028, 501, 1028, 490, 1027, 489});
    contourPointsData.push_back({1050, 198, 1049, 199, 1040, 199, 1040, 210, 1041, 211, 1040, 212, 1040, 214, 1045, 214,
                                 1046, 213, 1049, 213, 1050, 212, 1051, 212, 1052, 211, 1053, 211, 1054, 210, 1055, 210,
                                 1056, 209, 1058, 209, 1059, 208, 1059, 200, 1058, 200, 1057, 199, 1051, 199});
    //
    std::vector<std::vector<float>> openCV_minAreaRect_results;
    openCV_minAreaRect_results.push_back({868.0, 604.0, 845.0, 604.0, 845.0, 600.0, 868.0, 600.0});
    openCV_minAreaRect_results.push_back({962.0, 504.0, 962.0, 489.0, 1028.0, 489.0, 1028.0, 504.0});
    openCV_minAreaRect_results.push_back({1040.0, 214.0, 1040.0, 198.0, 1059.0, 198.0, 1059.0, 214.0});

    // point number in each contour
    nvcv::Tensor inPointNumInContour{
        nvcv::TensorShape{{1, batchsize}, nvcv::TENSOR_NW},
        nvcv::TYPE_S32
    };
    auto inPointNumInContourAccess    = nvcv::TensorDataAccessStrided::Create(inPointNumInContour.exportData());
    auto numPointNumInContourElements = inPointNumInContourAccess->sampleStride() / sizeof(int);
    std::vector<int> inPointNumInContourValues(numPointNumInContourElements, 0);

    for (int i = 0; i < batchsize; i++)
    {
        inPointNumInContourValues[i] = contourPointsData[i].size() / 2;
    }
    int maxPointsNumInCountour = *std::max_element(inPointNumInContourValues.begin(), inPointNumInContourValues.end());

    // inTensor
    auto tshapeIn = nvcv::TensorShape{
        {batchsize, maxPointsNumInCountour, 2},
        nvcv::TENSOR_NWC
    };
    nvcv::DataType dtypeIn = nvcv::TYPE_S16;
    nvcv::Tensor   inContours{tshapeIn, dtypeIn};
    auto           inContursAccess     = nvcv::TensorDataAccessStrided::Create(inContours.exportData());
    auto           numContoursElements = inContursAccess->sampleStride() / (2 * dtypeIn.strideBytes());

    // outTensor
    // 8 is the tl tr bl br cooridinates
    auto tshapeOut = nvcv::TensorShape{
        {batchsize, 8},
        nvcv::TENSOR_NW
    };
    auto                            dtypeOut = nvcv::TYPE_F32;
    nvcv::Tensor                    outMinAreaRect{tshapeOut, dtypeOut};
    std::vector<std::vector<float>> testVec(batchsize, std::vector<float>(8, 0));
    auto                            outAccess = nvcv::TensorDataAccessStrided::Create(outMinAreaRect.exportData());

    for (int i = 0; i < batchsize; i++)
    {