前段时间研究了下印象笔记的扫描宝,发现它将手机拍下的照片都能处理成白板背景,而且处理的很干净,出于好奇并学习,自己琢磨出了个简易的去底色算法,虽然不能跟印象笔记的相比,但是效果也还阔以,供自己学习了,有兴趣的朋友欢迎指教。
文中测试图分别如下:
图1
图2
图3
图4
法一:伽马变换
主要思想:调整色阶,明暗图像求加;
void ls::gamma2Add(cv::Mat &src,cv::Mat &dst)
{
Mat image;
resize(src, image, Size(((float)src.cols/(float)src.rows)*krows,krows));
Scan scan;
cv::Mat dst1,dst2;
scan.gammaCorrection(image, dst1, 0.45);
scan.gammaCorrection(image, dst2, 2);
dst = dst1 + dst2;
}
效果如下:
法二:替换背景的像素值;
主要思想,首先遍历图像(遍历可以优化),统计出图像中最多的颜色像素,然后替换成255;
void changeRGB(Mat &src)
{
// Mat image;
// detailEnhance(src, src);
Mat gray;
cvtColor(src, gray, CV_BGR2GRAY);
MatND hist = getHistogram(gray);
double maxVal = 0,minVal = 0;
//获取最大值和最小值
int hpt = static_cast<int>(0.9 * histSize[0]);
minMaxLoc(hist, &minVal, &maxVal,0,0);
cout<<minVal<<","<<maxVal<<endl;
vector<int> kChange;
// int th = me / maxVal;
// int nonzero = countNonZero(gray);
// double ratio = nonzero / (double)(gray.cols * gray.rows);
// cout<<ratio<<endl;
for (int i = 0; i < histSize[0]; i ++)
{
float binVal = hist.at<float>(i);
//cout<<binVal<<endl;
int intensity = static_cast<int>(binVal*hpt/maxVal);
if (abs(intensity - 255) < 60)
{
intensity = 255;
hist.at<float>(i) = intensity * maxVal/hpt;
}
// if (abs(intensity - 0) > 30) {
// intensity = 0;
// hist.at<float>(i) = intensity * maxVal/hpt;
// }
//
}
minMaxLoc(hist,&minVal, &maxVal,0,0);
cout<<minVal<<","<<maxVal<<endl;
for (int num = 0; num < histSize[0]; num ++)
{
if (hist.at<float>(num) >= 0.5*maxVal)//( hist.at<float>(i) >= ratio*maxVal && hist.at<float>(i) <= 1.0*maxVal)//3.63
{
kChange.push_back(num);
}
}
int nl = src.rows;
int nc = src.cols;
// if (src.isContinuous()) {
// nc = nc * nl;
// nl = 1;
// }
for (int j = 0; j < nl; j ++)
{
// uchar *data = src.ptr<uchar>(j);
for (int l = 0; l < nc; l ++)
{
int b = 0.11 * src.at<Vec3b>(j, l)[0];
int g = 0.59 * src.at<Vec3b>(j, l)[1];
int r = 0.30 * src.at<Vec3b>(j, l)[2];
int greyVal = b + g + r;
for (int ii = 0; ii < kChange.size(); ii ++)
{
if (abs(greyVal - kChange[ii]) <= 18)//
// if (greyVal == kChange[ii])
{
src.at<Vec3b>(j, l)[0] = 255;
src.at<Vec3b>(j, l)[1] = 255;
src.at<Vec3b>(j, l)[2] = 255;
}
}
}
}
}
效果如下:
不难发现还是有一些瑕疵在的,希望能与大家交流讨论。。