C++とOpenCVで学ぶハフ変換による直線と円の検出実践ガイド

#include <opencv2/opencv.hpp>
#include <vector>
#include <cmath>
int main() {
    // 画像をグレースケールで読み込みます
    cv::Mat grayImage = cv::imread("image.jpg", cv::IMREAD_GRAYSCALE);
    if (grayImage.empty()) {
        return -1;
    }
    // Cannyエッジ検出でエッジを抽出します
    cv::Canny(grayImage, grayImage, 50, 150);
    // ハフ変換で直線を検出します
    std::vector<cv::Vec2f> lines;
    cv::HoughLines(grayImage, lines, 1, CV_PI / 180, 100);
    // 結果表示のため、グレースケール画像をカラーに変換します
    cv::Mat colorImage;
    cv::cvtColor(grayImage, colorImage, cv::COLOR_GRAY2BGR);
    // 検出された直線を赤色の線で描画します
    for (size_t i = 0; i < lines.size(); i++) {
        float rho = lines[i][0];
        float theta = lines[i][1];
        double a = std::cos(theta);
        double b = std::sin(theta);
        double x0 = a * rho;
        double y0 = b * rho;
        cv::Point pt1(cvRound(x0 + 1000 * (-b)), cvRound(y0 + 1000 * (a)));
        cv::Point pt2(cvRound(x0 - 1000 * (-b)), cvRound(y0 - 1000 * (a)));
        cv::line(colorImage, pt1, pt2, cv::Scalar(0, 0, 255), 2, cv::LINE_AA);
    }
    // 結果のウィンドウを表示します
    cv::imshow("Detected Lines", colorImage);
    cv::waitKey(0);
    return 0;
}

#include <opencv2/opencv.hpp>
#include <vector>
int main() {
    // 画像をグレースケールで読み込みます
    cv::Mat grayImage = cv::imread("image.jpg", cv::IMREAD_GRAYSCALE);
    if (grayImage.empty()) {
        return -1;
    }
    // ガウシアンブラーでノイズを低減します
    cv::GaussianBlur(grayImage, grayImage, cv::Size(9, 9), 2, 2);
    // ハフ変換で円を検出します
    std::vector<cv::Vec3f> circles;
    cv::HoughCircles(grayImage, circles, cv::HOUGH_GRADIENT, 1, grayImage.rows / 8, 100, 30, 1, 30);
    // 結果表示のため、グレースケール画像をカラーに変換します
    cv::Mat colorImage;
    cv::cvtColor(grayImage, colorImage, cv::COLOR_GRAY2BGR);
    // 円の中心と輪郭を描画します
    for (size_t i = 0; i < circles.size(); i++) {
        cv::Point center(cvRound(circles[i][0]), cvRound(circles[i][1]));
        int radius = cvRound(circles[i][2]);
        // 中心を緑色の小さな円で描画します
        cv::circle(colorImage, center, 3, cv::Scalar(0, 255, 0), -1);
        // 外枠を赤色で描画します
        cv::circle(colorImage, center, radius, cv::Scalar(0, 0, 255), 3);
    }
    // 結果のウィンドウを表示します
    cv::imshow("Detected Circles", colorImage);
    cv::waitKey(0);
    return 0;
}