OpenCV: samples/cpp/create_mask.cpp

Logo

OpenCV 4.0.1

Open Source Computer Vision

[フレーム]

samples/cpp/create_mask.cpp

This program demonstrates using mouse events and how to make and use a mask image (black and white) .

* create_mask.cpp

* Author:

* Siddharth Kherada <siddharthkherada27[at]gmail[dot]com>

* This tutorial demonstrates how to make mask image (black and white).

* The program takes as input a source image and outputs its corresponding

* mask image.

#include "opencv2/imgproc.hpp"

#include "opencv2/imgcodecs.hpp"

#include "opencv2/highgui.hpp"

#include <iostream>

using namespace std;

using namespace cv;

Mat src, img1, mask, final;

Point point;

vector<Point> pts;

int drag = 0;

int var = 0;

int flag = 0;

void mouseHandler(int, int, int, int, void*);

void mouseHandler(int event, int x, int y, int, void*)

{

if (event == EVENT_LBUTTONDOWN && !drag)

{

if (flag == 0)

{

if (var == 0)

img1 = src.clone();

point = Point(x, y);

circle(img1, point, 2, Scalar(0, 0, 255), -1, 8, 0);

pts.push_back(point);

var++;

drag = 1;

if (var > 1)

line(img1,pts[var-2], point, Scalar(0, 0, 255), 2, 8, 0);

imshow("Source", img1);

}

if (event == EVENT_LBUTTONUP && drag)

{

imshow("Source", img1);

drag = 0;

}

if (event == EVENT_RBUTTONDOWN)

{

flag = 1;

img1 = src.clone();

if (var != 0)

{

polylines( img1, pts, 1, Scalar(0,0,0), 2, 8, 0);

}

imshow("Source", img1);

}

if (event == EVENT_RBUTTONUP)

{

flag = var;

final = Mat::zeros(src.size(), CV_8UC3);

mask = Mat::zeros(src.size(), CV_8UC1);

vector<vector<Point> > vpts;

vpts.push_back(pts);

fillPoly(mask, vpts, Scalar(255, 255, 255), 8, 0);

bitwise_and(src, src, final, mask);

imshow("Mask", mask);

imshow("Result", final);

imshow("Source", img1);

}

if (event == EVENT_MBUTTONDOWN)

{

pts.clear();

var = 0;

drag = 0;

flag = 0;

imshow("Source", src);

}

int main(int argc, char **argv)

{

CommandLineParser parser(argc, argv, "{@input | lena.jpg | input image}");

parser.about("This program demonstrates using mouse events\n");

parser.printMessage();

cout << "\n\tleft mouse button - set a point to create mask shape\n"

"\tright mouse button - create mask from points\n"

"\tmiddle mouse button - reset\n";

String input_image = parser.get<String>("@input");

src = imread(samples::findFile(input_image));

if (src.empty())

{

printf("Error opening image: %s\n", input_image.c_str());

return 0;

}

namedWindow("Source", WINDOW_AUTOSIZE);

setMouseCallback("Source", mouseHandler, NULL);

imshow("Source", src);

waitKey(0);

return 0;

}

Generated on Sat Dec 22 2018 08:24:04 for OpenCV by doxygen 1.8.12