首页 > 代码库 > [OpenCV] Samples 07: create_mask
[OpenCV] Samples 07: create_mask
鼠标画线,圈地,构造相关mask图片(黑白)。
支持鼠标左键右键中间键点击事件。
/** 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/imgcodecs.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*){ // Jeff --> Click Left button: status down. 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); } } // Jeff --> Click Left button: status up. if (event == EVENT_LBUTTONUP && drag) { imshow("Source", img1); drag = 0; } /* * Jeff --> Click Right button. */ 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){ cv::CommandLineParser parser(argc, argv, "{help h | | show help message}{@input | | input image}"); if (parser.has("help")) { parser.printMessage(); return 0; } string input_image = parser.get<string>("@input"); if (input_image.empty()) { parser.printMessage(); parser.printErrors(); return 0; } Mat src = http://www.mamicode.com/imread(input_image);"channel = " << channel << endl; res1 = Mat::zeros(img0.size(),CV_8UC1); final = Mat::zeros(img0.size(),CV_8UC3); //////////// source image /////////////////// namedWindow("Source", 1); /* * Jeff --> listen to mouse event. */ setMouseCallback("Source", mouseHandler, NULL); imshow("Source", img0); waitKey(0); img0.release(); img1.release();}
[OpenCV] Samples 07: create_mask
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。