原创的一套基于opencv的图片增强代码
增强方式有四种,包括了直方图均衡化、拉普拉斯、Log、Gamma,下面上带注释的代码
#include <opencv2/opencv.hpp>
#include <iostream>
#include <fstream>
#include <string>
using namespace std;
using namespace cv;
int main(){
Mat image; //图像变量image
string ImgName; //图像名ImgName
ifstream infile("/home/tao/laplace_transform2/SHUJU.txt");//读取该路径下的记录图片名的txt文件夹,
// 保存在infile中
while (getline(infile, ImgName))//getline是读取文件中的行,以换行符分段,一次循环读一行
{
ImgName = ImgName.substr(0, ImgName.length() - 1);//因为txt中的换行符会被认为是字符\r,影响读取,
// 截掉最后两位
cout << "Handle" << ImgName << endl;//输出处理信号
ImgName = "/home/tao/laplace_transform2/shuju/" + ImgName ;//给ImgName加上文件夹路径,
//文件夹路径+文件名才能用imread导入图片
image = imread(ImgName);//导入图片到image
if (!image.empty())
{ //增强处理,最后得到的图片为imageEnhance
Mat imageEnhance;
Mat kernel = (Mat_<float>(3, 3) << 0, -1, 0, 0, 5, 0, 0, -1, 0);
filter2D(image, imageEnhance, CV_8UC3, kernel);
imwrite(ImgName,imageEnhance);//imwrite函数保存至原路径将原图覆盖
cout << "Done" << endl;
}
}
cvWaitKey(0);
return 0;
}
以上就是一个拉普拉斯增强代码,其他三种方式只要改动增强处理部分即可,稍微改一下变量大家都会
具体增强方法可参考以下链接
(https://www.cnblogs.com/jukan/p/7815722.html)
txt文件中的图片名要这种方式命名,回车分段,制作txt时可以用excel