照片背景色一键替换(简陋版)

照片背景色一键替换(简陋版)

这个程序是帮舍友替换证件照背景色的时候写的,可以用在图像质量要求不高的场景下,例如上传到招聘网站这种;
有一个存在的问题是证件照的背景可能影响头发边缘的颜色,这个程序并没有对这一点作出优化;
代码如下:

#include<iostream>
#include<opencv2/opencv.hpp>
using namespace std;
using namespace cv;

int dif = 40;//设置容差为10

int main(int argc, char* argv)
{
	Mat tmpImg = imread("test.jpg");
	int r, g, b;
	cout << "请输入目标背景色RGB,以空格为间隔\n白色为255 255 255" << endl;
	int r1, g1, b1;
	cin >> r1 >> g1 >> b1;

	r = tmpImg.at<Vec3b>(10, 10)[2];
	g = tmpImg.at<Vec3b>(10, 10)[1];
	b = tmpImg.at<Vec3b>(10, 10)[0];
	for (int i = 0; i < tmpImg.rows; i++) {
		for (int j = 0; j < tmpImg.cols; j++) {
			if ((abs(tmpImg.at<Vec3b>(i, j)[2] - r)<=dif) &&
				(abs(tmpImg.at<Vec3b>(i, j)[1] - g) <= dif) &&
				(abs(tmpImg.at<Vec3b>(i, j)[0] - b) <= dif)) {
				tmpImg.at<Vec3b>(i, j)[2] = r1;
				tmpImg.at<Vec3b>(i, j)[1] = g1;
				tmpImg.at<Vec3b>(i, j)[0] = b1;
			}
		}
	}

	imwrite("test2.jpg", tmpImg);

	return 0;
}

分享:

扫一扫在手机阅读、分享本文

评论