mirror of
https://github.com/opencv/opencv.git
synced 2024-11-25 03:30:34 +08:00
f85a615f3f
Update 1 update 2 Update 3 Update 4 Update 5 update 6 Update 7 Update 8 Update 9 Update 10 Update 11 Update 12 Update 13 Update 14 Update 15 Update 16 Update 17 Update 18 Update 19 Update 20 Update 21 Update 22 Removed Headers Update 23 Update 24 Update 25 Update 26 Update 27 Update 28
152 lines
3.2 KiB
C++
152 lines
3.2 KiB
C++
/*
|
|
* 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 ouputs its corresponding
|
|
* mask image.
|
|
*/
|
|
|
|
#include "opencv2/imgproc.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
#include "opencv2/core.hpp"
|
|
#include <iostream>
|
|
#include <stdlib.h>
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
|
|
Mat img0, img1, res1, final;
|
|
|
|
Point point;
|
|
int drag = 0;
|
|
|
|
int numpts = 100;
|
|
Point* pts = new Point[100];
|
|
|
|
int var = 0;
|
|
int flag = 0;
|
|
int flag1 = 0;
|
|
|
|
int minx,miny,maxx,maxy,lenx,leny;
|
|
|
|
int channel;
|
|
|
|
void mouseHandler(int, int, int, int, void*);
|
|
|
|
void mouseHandler(int event, int x, int y, int, void*)
|
|
{
|
|
|
|
if (event == EVENT_LBUTTONDOWN && !drag)
|
|
{
|
|
if(flag1 == 0)
|
|
{
|
|
if(var==0)
|
|
img1 = img0.clone();
|
|
point = Point(x, y);
|
|
circle(img1,point,2,Scalar(0, 0, 255),-1, 8, 0);
|
|
pts[var] = 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)
|
|
{
|
|
flag1 = 1;
|
|
img1 = img0.clone();
|
|
for(int i = var; i < numpts ; i++)
|
|
pts[i] = point;
|
|
|
|
if(var!=0)
|
|
{
|
|
const Point* pts3[1] = {&pts[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);
|
|
}
|
|
|
|
if (event == EVENT_RBUTTONUP)
|
|
{
|
|
flag = var;
|
|
|
|
final = Mat::zeros(img0.size(),CV_8UC3);
|
|
res1 = Mat::zeros(img0.size(),CV_8UC1);
|
|
const Point* pts4[1] = {&pts[0]};
|
|
|
|
fillPoly(res1, pts4,&numpts, 1, Scalar(255, 255, 255), 8, 0);
|
|
bitwise_and(img0, img0, final,res1);
|
|
imshow("mask",res1);
|
|
imwrite("mask.png",res1);
|
|
|
|
imshow("Source", img1);
|
|
|
|
}
|
|
if (event == EVENT_MBUTTONDOWN)
|
|
{
|
|
for(int i = 0; i < numpts ; i++)
|
|
{
|
|
pts[i].x=0;
|
|
pts[i].y=0;
|
|
}
|
|
var = 0;
|
|
flag1 = 0;
|
|
minx = INT_MAX; miny = INT_MAX; maxx = INT_MIN; maxy = INT_MIN;
|
|
imshow("Source", img0);
|
|
drag = 0;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
|
|
if(argc != 2)
|
|
{
|
|
cout << "usage: " << argv[0] << " <input_image>" << endl;
|
|
exit(1);
|
|
}
|
|
|
|
Mat src = imread(argv[1]);
|
|
|
|
minx = INT_MAX; miny = INT_MAX; maxx = INT_MIN; maxy = INT_MIN;
|
|
|
|
img0 = src;
|
|
|
|
channel = img0.channels();
|
|
|
|
res1 = Mat::zeros(img0.size(),CV_8UC1);
|
|
final = Mat::zeros(img0.size(),CV_8UC3);
|
|
//////////// source image ///////////////////
|
|
|
|
namedWindow("Source", 1);
|
|
setMouseCallback("Source", mouseHandler, NULL);
|
|
imshow("Source", img0);
|
|
waitKey(0);
|
|
|
|
img0.release();
|
|
img1.release();
|
|
}
|