modules/features2d/test/test_blobdetector.cpp - 4.x - Alx89/OpenCV

modules/features2d/test/test_blobdetector.cpp

46 строк2 KB

Petr Glotov

Merge pull request #21942 from pglotov:add-blob-contours

07 окт 2022, 19:07

07 окт 2022, 19:07a3ebafb

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.

#include "test_precomp.hpp"

namespace opencv_test { namespace {
TEST(Features2d_BlobDetector, bug_6667)
{
    cv::Mat image = cv::Mat(cv::Size(100, 100), CV_8UC1, cv::Scalar(255, 255, 255));
    cv::circle(image, Point(50, 50), 20, cv::Scalar(0), -1);
    SimpleBlobDetector::Params params;
    params.minThreshold = 250;
    params.maxThreshold = 260;
    params.minRepeatability = 1;  // https://github.com/opencv/opencv/issues/6667
    std::vector<KeyPoint> keypoints;

    Ptr<SimpleBlobDetector> detector = SimpleBlobDetector::create(params);
    detector->detect(image, keypoints);
    ASSERT_NE((int) keypoints.size(), 0);
}

TEST(Features2d_BlobDetector, withContours)
{
    cv::Mat image = cv::Mat(cv::Size(100, 100), CV_8UC1, cv::Scalar(255, 255, 255));
    cv::circle(image, Point(50, 50), 20, cv::Scalar(0), -1);
    SimpleBlobDetector::Params params;
    params.minThreshold = 250;
    params.maxThreshold = 260;
    params.minRepeatability = 1; // https://github.com/opencv/opencv/issues/6667
    params.collectContours = true;
    std::vector<KeyPoint> keypoints;

    Ptr<SimpleBlobDetector> detector = SimpleBlobDetector::create(params);
    detector->detect(image, keypoints);
    ASSERT_NE((int)keypoints.size(), 0);

    ASSERT_GT((int)detector->getBlobContours().size(), 0);
    std::vector<Point> contour = detector->getBlobContours()[0];
    ASSERT_TRUE(std::any_of(contour.begin(), contour.end(),
                            [](Point p)
                            {
                                return abs(p.x - 30) < 2 && abs(p.y - 50) < 2;
                            }));
}
}} // namespace