modules/features2d/perf/perf_feature2d.cpp - 4.x - Alx89/OpenCV

modules/features2d/perf/perf_feature2d.cpp
72 строки2 KB
Alexander Alekhin
ts: refactor OpenCV tests
03 фев 2018, 22:39
03 фев 2018, 22:394a297a2
#include "perf_feature2d.hpp"

namespace opencv_test
{
using namespace perf;

PERF_TEST_P(feature2d, detect, testing::Combine(Feature2DType::all(), TEST_IMAGES))
{
    Ptr<Feature2D> detector = getFeature2D(get<0>(GetParam()));
    std::string filename = getDataPath(get<1>(GetParam()));
    Mat img = imread(filename, IMREAD_GRAYSCALE);

    ASSERT_FALSE(img.empty());
    ASSERT_TRUE(detector);

    declare.in(img);
    Mat mask;
    vector<KeyPoint> points;

    TEST_CYCLE() detector->detect(img, points, mask);

    EXPECT_GT(points.size(), 20u);
    SANITY_CHECK_NOTHING();
}

PERF_TEST_P(feature2d, extract, testing::Combine(testing::Values(DETECTORS_EXTRACTORS), TEST_IMAGES))
{
    Ptr<Feature2D> detector = AKAZE::create();
    Ptr<Feature2D> extractor = getFeature2D(get<0>(GetParam()));
    std::string filename = getDataPath(get<1>(GetParam()));
    Mat img = imread(filename, IMREAD_GRAYSCALE);

    ASSERT_FALSE(img.empty());
    ASSERT_TRUE(extractor);

    declare.in(img);
    Mat mask;
    vector<KeyPoint> points;
    detector->detect(img, points, mask);

    EXPECT_GT(points.size(), 20u);

    Mat descriptors;

    TEST_CYCLE() extractor->compute(img, points, descriptors);

    EXPECT_EQ((size_t)descriptors.rows, points.size());
    SANITY_CHECK_NOTHING();
}

PERF_TEST_P(feature2d, detectAndExtract, testing::Combine(testing::Values(DETECTORS_EXTRACTORS), TEST_IMAGES))
{
    Ptr<Feature2D> detector = getFeature2D(get<0>(GetParam()));
    std::string filename = getDataPath(get<1>(GetParam()));
    Mat img = imread(filename, IMREAD_GRAYSCALE);

    ASSERT_FALSE(img.empty());
    ASSERT_TRUE(detector);

    declare.in(img);
    Mat mask;
    vector<KeyPoint> points;
    Mat descriptors;

    TEST_CYCLE() detector->detectAndCompute(img, mask, points, descriptors, false);

    EXPECT_GT(points.size(), 20u);
    EXPECT_EQ((size_t)descriptors.rows, points.size());
    SANITY_CHECK_NOTHING();
}

} // namespace