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