opencv通过鼠标生成掩码区域的代码实现

文章主要思路是通过鼠标点击两点的方式来生成掩码矩形，掩码圆。本文例子鼠标点击两点生成矩形框来选择保存区域。文章中有注释掉的 删除区域、圆掩码的代码。

代码

#include<opencv2/opencv.hpp>
#include<opencv2/imgcodecs.hpp>
#include<opencv2/highgui.hpp>
#include<opencv2/imgproc.hpp>
#include "opencv2/imgcodecs.hpp"
#include<iostream>

using namespace std;
using namespace cv;

vector<Point> mousePoints;
Point points;
void on_mouse(int EVENT, int x, int y, int flags, void* userdata)
{

    Mat hh;
    hh = *(Mat*)userdata;
    Point p(x, y);
    switch (EVENT)
    {
    case EVENT_LBUTTONDOWN:
    {
        points.x = x;
        points.y = y;
        mousePoints.push_back(points);

        circle(hh, points, 1, Scalar(255, 255, 255), -1);
        imshow("mouseCallback", hh);
    }
    break;
    }

}

int selectPolygon(cv::Mat srcMat, cv::Mat& dstMat)
{
    vector<vector<Point>> contours;
    cv::Mat selectMat;
    cv::Mat m = cv::Mat::zeros(srcMat.size(), CV_8UC1);
    m = 0;

    if (!srcMat.empty()) {
        srcMat.copyTo(selectMat);

    }
    else {
        std::cout << "failed to read image!:" << std::endl;
        return -1;
    }
    //该程序只能两点选取 
    //左键选取两点后  回车键
    namedWindow("mouseCallback");
    imshow("mouseCallback", selectMat);
    setMouseCallback("mouseCallback", on_mouse, &selectMat);
    waitKey(0);
    destroyAllWindows();
    //计算roi  本文只能按照两点矩形方式生成掩码
    contours.push_back(mousePoints);
    if (contours[0].size() != 2) {
        std::cout << "failed to read image!:" << std::endl;
        return -1;
    }
    cv::rectangle(m, contours[0].at(0), contours[0][1], cv::Scalar(255), -1, 8, 0);
    //下例m=255 删除区域
    /*m=255;  
    cv::rectangle(m, contours[0].at(0), contours[0][1], cv::Scalar(0), -1, 8, 0);*/

    /*下例可以生成掩码圆区域 也是两点生成方式
    int r = (int)sqrt((contours[0][1].x - contours[0][0].x) * (contours[0][1].x - contours[0][0].x) + (contours[0][1].y - contours[0][0].y) * (contours[0][1].y - contours[0][0].y));
    cv::circle(m, contours[0][0], r, cv::Scalar(0), -1, 8, 0);*/
    

    m.copyTo(dstMat);

    return 0;
}

int main()
{
    Mat src = imread("C:/Users/17482/Desktop/222.jpg");
    Mat out, dst;
    int a = selectPolygon(src,out);
    if (a != 0)
    {
        cout << "原图路径不对" << endl;
    }
    else
    {
        bitwise_and(src, src, dst, out);
        namedWindow("原图");
        moveWindow("原图", 0, 0);
        imshow("原图", src);
        imshow("掩码图", out);
        imshow("结果图", dst);
        waitKey(0);
    

    }

    system("pause");
    return 0;
}