OpenCV mask掩膜

今天在做portrait-matting算法demo时,想着换张背景图试试看,直接操作像素内存buffer当然没有问题(以前写过太多直接改像素值的操作,也没有什么技术含量,重复造轮子,写的想吐了),于是,试着调用OpenCV库函数完成功能,减少工作量。
其实,换背景的实质就是图层mix,可以用到OpenCV mask掩膜,如果mask是非二值化的,用矩阵运算比较合适,可以确保过渡自然。下面说下如何使用mask掩膜。

作为掩膜mask的数据类型必须是CV_8U。

通常用法(mask掩膜)
Mat src=imread("example.jpg");
Rect t_rect(100,100,200,200);	//ROI
Mat mask=zeros(src.size(),CV_8UC1);	
mask(t_rect).setTo(255);
Mat t_res_image1 = src(t_rect);	//抠出ROI
Mat t_res_image2;
src.copyTo(t_res_image2,mask);	//只拷贝mask像素值为255的区域像素
换背景用法(非mask掩膜)
Mat bg;		//背景
Mat fg;		//前景
Mat mask;	//matting mask (0~255 灰度图,single channel)
cvtColor(mask,mask,COLOR_GRAY2BGR);
Mat mask2;
mask.convertTo(mask2,CV_32FC3,1.0/255.0,0);		//数据类型转换,mask2 = mask/255.0
Mat frame = fg;
Mat frame2;
frame.convertTo(frame2,CV_32F,1.0,0);	// 数据类型转换
Mat frame3 = frame2.mul(mask2)+bg.mul(Mat(mask2.size(),CV_32FC3,Scalar(1,1,1))-mask2);
frame3.convertTo(frame2,CV_8UC3,1,0);
imshow("result",frame2);

猜你喜欢

转载自blog.csdn.net/zxcasd11/article/details/107005763