mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-21 19:07:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #12036 from sturkmen72:update_create_mask_cpp

Browse Source
This commit is contained in:
Alexander Alekhin 2018-07-31 15:02:16 +00:00
commit 17196bb4fa
1 changed files with 42 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

126
samples/cpp/create_mask.cpp
View File

@ -12,26 +12,18 @@
#include "opencv2/imgproc.hpp" #include "opencv2/imgproc.hpp"
#include "opencv2/imgcodecs.hpp" #include "opencv2/imgcodecs.hpp"
#include "opencv2/highgui.hpp" #include "opencv2/highgui.hpp"
#include "opencv2/core.hpp"
#include <iostream> #include <iostream>
#include <stdlib.h>
using namespace std; using namespace std;
using namespace cv; using namespace cv;
Mat img0, img1, res1, final; Mat src, img1, mask, final;
Point point; Point point;
vector<Point> pts;
int drag = 0; int drag = 0;
int numpts = 100;
Point* pts = new Point[100];
int var = 0; int var = 0;
int flag = 0; int flag = 0;
int flag1 = 0;
int minx,miny,maxx,maxy,lenx,leny;
void mouseHandler(int, int, int, int, void*); void mouseHandler(int, int, int, int, void*);
@ -40,16 +32,17 @@ void mouseHandler(int event, int x, int y, int, void*)
if (event == EVENT_LBUTTONDOWN && !drag) if (event == EVENT_LBUTTONDOWN && !drag)
{ {
if(flag1 == 0) if (flag == 0)
{ {
if(var==0) if (var == 0)
img1 = img0.clone(); img1 = src.clone();
point = Point(x, y); point = Point(x, y);
circle(img1,point,2,Scalar(0, 0, 255),-1, 8, 0); circle(img1, point, 2, Scalar(0, 0, 255), -1, 8, 0);
pts[var] = point; pts.push_back(point);
var++; var++;
drag = 1; drag = 1;
if(var>1)
if (var > 1)
line(img1,pts[var-2], point, Scalar(0, 0, 255), 2, 8, 0); line(img1,pts[var-2], point, Scalar(0, 0, 255), 2, 8, 0);
imshow("Source", img1); imshow("Source", img1);
@ -59,103 +52,68 @@ void mouseHandler(int event, int x, int y, int, void*)
if (event == EVENT_LBUTTONUP && drag) if (event == EVENT_LBUTTONUP && drag)
{ {
imshow("Source", img1); imshow("Source", img1);
drag = 0; drag = 0;
} }
if (event == EVENT_RBUTTONDOWN) if (event == EVENT_RBUTTONDOWN)
{ {
flag1 = 1; flag = 1;
img1 = img0.clone(); img1 = src.clone();
for(int i = var; i < numpts ; i++)
pts[i] = point;
if(var!=0) if (var != 0)
{ {
const Point* pts3[1] = {&pts[0]}; polylines( img1, pts, 1, Scalar(0,0,0), 2, 8, 0);
polylines( img1, pts3, &numpts,1, 1, Scalar(0,0,0), 2, 8, 0);
} }
for(int i=0;i<var;i++)
{
minx = min(minx,pts[i].x);
maxx = max(maxx,pts[i].x);
miny = min(miny,pts[i].y);
maxy = max(maxy,pts[i].y);
}
lenx = maxx - minx;
leny = maxy - miny;
imshow("Source", img1); imshow("Source", img1);
} }
if (event == EVENT_RBUTTONUP) if (event == EVENT_RBUTTONUP)
{ {
flag = var; flag = var;
final = Mat::zeros(src.size(), CV_8UC3);
mask = Mat::zeros(src.size(), CV_8UC1);
final = Mat::zeros(img0.size(),CV_8UC3); vector<vector<Point> > vpts;
res1 = Mat::zeros(img0.size(),CV_8UC1); vpts.push_back(pts);
const Point* pts4[1] = {&pts[0]}; fillPoly(mask, vpts, Scalar(255, 255, 255), 8, 0);
bitwise_and(src, src, final, mask);
fillPoly(res1, pts4,&numpts, 1, Scalar(255, 255, 255), 8, 0); imshow("Mask", mask);
bitwise_and(img0, img0, final,res1); imshow("Result", final);
imshow("mask",res1);
imwrite("mask.png",res1);
imshow("Source", img1); imshow("Source", img1);
} }
if (event == EVENT_MBUTTONDOWN) if (event == EVENT_MBUTTONDOWN)
{ {
for(int i = 0; i < numpts ; i++) pts.clear();
{
pts[i].x=0;
pts[i].y=0;
}
var = 0; var = 0;
flag1 = 0;
minx = INT_MAX; miny = INT_MAX; maxx = INT_MIN; maxy = INT_MIN;
imshow("Source", img0);
drag = 0; drag = 0;
flag = 0;
imshow("Source", src);
} }
} }
static void help()
{
cout << "\nThis program demonstrates using mouse events"
"\nCall:\n"
"./create_mask <image_name>\n"
"\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" << endl;
}
int main(int argc, char **argv) int main(int argc, char **argv)
{ {
cv::CommandLineParser parser(argc, argv, "{@input | ../data/lena.jpg | input image}"); CommandLineParser parser(argc, argv, "{@input | ../data/lena.jpg | input image}");
help(); parser.about("This program demonstrates using mouse events\n");
string input_image = parser.get<string>("@input"); parser.printMessage();
if (input_image.empty()) cout << "\n\tleft mouse button - set a point to create mask shape\n"
{ "\tright mouse button - create mask from points\n"
parser.printMessage(); "\tmiddle mouse button - reset\n";
parser.printErrors(); String input_image = parser.get<String>("@input");
src = imread(input_image);
if (src.empty())
{
printf("Error opening image: %s\n", input_image.c_str());
return 0; return 0;
} }
Mat src = imread(input_image); namedWindow("Source", WINDOW_AUTOSIZE);
minx = INT_MAX; miny = INT_MAX; maxx = INT_MIN; maxy = INT_MIN;
img0 = src;
res1 = Mat::zeros(img0.size(),CV_8UC1);
final = Mat::zeros(img0.size(),CV_8UC3);
//////////// source image ///////////////////
namedWindow("Source", 1);
setMouseCallback("Source", mouseHandler, NULL); setMouseCallback("Source", mouseHandler, NULL);
imshow("Source", img0); imshow("Source", src);
waitKey(0); waitKey(0);
return 0; return 0;