OpenCV 绘制轮廓

最新推荐文章于 2024-10-14 22:55:06 发布

侃侃_天下

最新推荐文章于 2024-10-14 22:55:06 发布

阅读量452

点赞数 5

文章标签： opencv 人工智能计算机视觉

本文链接：https://blog.csdn.net/douzi949389/article/details/142166598

版权


#include <opencv2/opencv.hpp>
#include <vector>

// 全局变量
cv::Mat src, copy, dst, gray;
std::vector<std::vector<cv::Point>> contours;
//std::vector<cv::Point> lines;
std::vector<cv::Point> lines;
bool drawing = false;
cv::Point prevPoint;
cv::Point firstPoint;
// 鼠标回调函数
static void mouseHandler(int event, int x, int y, int flags, void* userData) {
    if (event == cv::EVENT_LBUTTONDOWN) {
        drawing = true;
        std::vector<cv::Point>().swap(lines);
        prevPoint = cv::Point(x, y);
        firstPoint = cv::Point(x, y);
        lines.push_back(prevPoint);
        
    }
    else if (event == cv::EVENT_MOUSEMOVE && flags == cv::EVENT_FLAG_LBUTTON) {
        if (drawing) {
            cv::Point pt(x, y);
            cv::line(dst, prevPoint, pt, cv::Scalar(255, 0, 0), 1, 8, 0);
            cv::line(gray, prevPoint, pt, cv::Scalar(255, 255, 255), 1, 8, 0);
            prevPoint = pt;
            lines.push_back(pt);
            
        }
    }
    else if (event == cv::EVENT_LBUTTONUP) {
        drawing = false;
        cv::Point pt(x, y);
        cv::line(dst, prevPoint, pt, cv::Scalar(255, 0, 0), 1, 8, 0);
        cv::line(gray, prevPoint, pt, cv::Scalar(255, 255, 255), 1, 8, 0);
        cv::line(dst, pt, firstPoint, cv::Scalar(255, 0, 0), 1, 8, 0);
        cv::line(gray, pt, firstPoint, cv::Scalar(255, 255, 255), 1, 8, 0);
        lines.push_back(pt);
        contours.push_back(lines);

    }
}
std::vector<cv::Scalar> colors =
{
    cv::Scalar(0, 0, 255),
    cv::Scalar(0, 255, 0),
    cv::Scalar(255, 0, 0),
    cv::Scalar(255, 255, 0),
    cv::Scalar(0, 255, 255),
    cv::Scalar(255, 0, 255)
};
int main() {
    std::string path = "D:/500_1s.png";
    src = cv::imread(path); // 替换为你的图片路径
    cv::resize(src, src, cv::Size(600, 600));
    dst = src.clone();
    copy = src.clone();

    cv::cvtColor(dst, gray, cv::COLOR_BGR2GRAY);
    gray.setTo(cv::Scalar(0));

    cv::namedWindow("Drawing");
    cv::setMouseCallback("Drawing", mouseHandler, 0);

    while (true) {
        cv::imshow("Drawing", dst);
        cv::imshow("gray", gray);
        
        int key = cv::waitKey(1);
        if (key == 'r') { // 按 'r' 重置
            dst = src.clone();
            contours.clear();
            gray.setTo(cv::Scalar(0));
        }
        else if (key == 27) { // 按 'ESC' 退出
            break;
        }

        if (!contours.empty())
        {
            for (int i = 0; i < contours.size(); ++i)
            {
                std::vector<std::vector<cv::Point>> ct;
                ct.push_back(contours[i]);
                cv::Scalar color = colors[i% colors.size()];
                std::string contour_name = "contours_" + std::to_string(i);
                cv::Mat contourImage = cv::Mat::zeros(gray.size(), CV_8UC1);
                cv::drawContours(contourImage, ct, -1, cv::Scalar(255, 255, 255), -1); // -1 表示填充轮廓内部
                std::vector<cv::Point> nonZeroCoords;
                cv::findNonZero(contourImage, nonZeroCoords);
                for (int i = 0; i < nonZeroCoords.size(); ++i)
                {
                    cv::circle(copy, nonZeroCoords[i], 1, color, cv::FILLED);
                }
            }
            cv::imshow("copy", copy);
        }
    }

    return 0;
}