OpenCV消除高亮illuminationChange函数的使用

学更好的别人,

做更好的自己。

——《微卡智享》

OpenCV消除高亮illuminationChange函数的使用_第1张图片

本文长度为1129,预计阅读4分钟

导语

上一篇《OpenCV极坐标变换函数warpPolar的使用》中介绍了极坐标变换的使用,文中提到过因为手机拍的照片,部分地方反光厉害。OpenCV本身也有一个消除高亮的函数,今天这篇就是来了解一下消除高亮函数的使用,就结果来说,有效果,但不多。

773d45bbf295e56eaf31cca6869dfc88.png

实现效果

OpenCV消除高亮illuminationChange函数的使用_第2张图片

OpenCV消除高亮illuminationChange函数的使用_第3张图片

OpenCV消除高亮illuminationChange函数的使用_第4张图片

OpenCV消除高亮illuminationChange函数的使用_第5张图片

从上几张图中可以看出,中间印的黑字的效果还明显一些,本身白字的和原来是差不多的。接下来就先说说去除高光函数illuminationChange。

illuminationChange函数

CV_EXPORTS_W void illuminationChange(InputArray src, InputArray mask, OutputArray dst,
        float alpha = 0.2f, float beta = 0.4f);

OpenCV消除高亮illuminationChange函数的使用_第6张图片

参数说明:

  • src——源图像,3通道图像

  • mask——掩膜,1通道或3通道图像都可以

  • dst——生成图像,同src

  • alpha、beta——两个参数共同决定消除高光后图像的模糊程度(范围0~2)beta越大,图片平滑越多,alpha越大,越接近原图细节。

2b93f054bba6fde75446b27180de9773.png

去高光函数比较简单,其实最核心的就是怎么样获取到掩膜,然后通过掩膜生成区域来实现去高光。具体的实现思路如下:

# 去高光实现步骤
1 获取图像区域,替换背景
2 二值化检测高光区域
3 查找高光区域轮廓生成掩膜
4 进行去高光操作

其实上面的步骤来说,在去高光操作最核心的两个步骤就是通过二值化检测高光区域,和查找轮廓后生成掩膜。结合我们上一篇的代码,看看怎么样实现的。

01

获取图像区域替换背景

在我们上一篇的代码基础上,把获取圆的区域这块单独写了个函数出来,目的主要是检测到圆后,将圆的区域截取出后,因为要去高光,所以圆外对我们来说是无用区域,直接填充到黑色,这样就减少掩膜查找无用高光区域了。

OpenCV消除高亮illuminationChange函数的使用_第7张图片

原理就是根据生成的矩形大小生成一个全黑的背景,然后将圆区域整个填充为白色,通过Mat.copyTo的方法实现。

OpenCV消除高亮illuminationChange函数的使用_第8张图片

02

查找高光区域

OpenCV消除高亮illuminationChange函数的使用_第9张图片

查找高光区域这里我也写成了一个函数,里面加了一步直方图均衡化,主要是本身图像清晰度也不高,所以使用直方图均衡化把图像对比度进行调整了一下,增强局部的对比度,更好的进行查找。

OpenCV消除高亮illuminationChange函数的使用_第10张图片

二值化后就是掩膜效果,为什么还要查找轮廓?

A

看到这里可能有人会提问,代码中threshold二值化找到的高光部分是不是直接可以当掩膜了,这个我开始也想直接这样的,省去了查找轮廓的部分,不过测试过程中直接报错了,也就是说去高光函数中掩膜区域里面只能是矩形,所以需要再加一步查找轮廓,针对轮廓生成外接矩形填充后才能使用

78bf7e20200e7f1fa9b02f91c9f0e01d.png

经过上面两步,去高光操作就实现了,后面的就是原图和处理后的图进行的极坐标变换对比。

完整代码

#include 
#include 


using namespace std;
using namespace cv;


//显示窗口设置  
//参数  img 显示的图像源,
//      winname 显示的窗口名称,
//      pointx  显示的坐标x
//      pointy  显示的坐标y
void setshowwindow(Mat img, string winname, int pointx, int pointy)
{
  //设置显示窗口
  namedWindow(winname, WindowFlags::WINDOW_NORMAL);
  //设置图像显示大小
  resizeWindow(winname, img.size());
  //设置图像显示位置
  moveWindow(winname, pointx, pointy);
}




//极坐标变换
//参数 flags=INTER_LINEAR 双线性插值
Mat warpPolarMat(Mat src, int flags = INTER_LINEAR + WARP_POLAR_LINEAR) {
  // 圆心坐标
  Point2f center = Point2f(src.cols / 2, src.rows / 2);
  // 圆的半径
  double maxRadius = min(center.y, center.x) - 1;
  // 圆的周长
  int circumference = maxRadius * 2 * 3.14;


  //输出图像
  Mat dst;
  // 极坐标变换, Size()表示OpenCV根据输入自行决定输出图像尺寸
  warpPolar(src, dst, Size(0, 0), center, maxRadius, flags);


  // 改变结果方向
  rotate(dst, dst, ROTATE_90_COUNTERCLOCKWISE);


  return dst;
}


//截取圆形图像,背景改为全黑
Mat copyCircleRoi(Mat src, Point center, int radius) {
  Mat rectsrc;
  //根据圆点和半径生成矩形
  Rect rect = Rect(Point(center.x - radius, center.y - radius), Point(center.x + radius, center.y + radius));
  //截图到当前圆的图像
  Mat rectroi = src(rect);


  //截取圆形区域
  Mat circleroi = Mat::zeros(rectroi.size(), CV_8U);
  //绘制检测到的圆
  circle(circleroi, Point(circleroi.rows / 2, circleroi.cols / 2), circleroi.rows / 2, Scalar(255), -1);


  //复制截取圆形区域
  rectroi.copyTo(rectsrc, circleroi);


  return rectsrc;
}


//去除高光
Mat matilluminationChange(Mat src) {
  Mat gray, threshmat, dst;
  //复制出来改为灰度图
  cvtColor(src, gray, COLOR_BGR2GRAY);


  //直方图均衡化
  equalizeHist(gray, gray);
  //imshow("equalizeHist", gray);


  //二值化操作,定义大于210的即为高光
  threshold(gray, threshmat, 210, 255, THRESH_BINARY);


  //查找图片中高亮区域轮廓
  vector > contours;
  findContours(threshmat, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);


  Mat mask = Mat::zeros(src.size(), CV_8UC1);
  for (int i = 0; i < contours.size(); ++i) {
    Rect rect = boundingRect(contours[i]);
    rectangle(mask, rect, Scalar(255), -1);
  }


  //imshow("mask", mask);


  //去高光
  illuminationChange(src, mask, dst, 1.0f, 0.1f);


  return dst;
}


int main(int argc, char** argv) {


  //测试图片文件  
  string testfile = "E:/DCIM/imagetest/06.png";


  //读取图片
  Mat src = imread(testfile);
  //修改图片大小
  setshowwindow(src, "src", 50, 200);
  imshow("src", src);




  //灰度图
  Mat gray;
  cvtColor(src, gray, COLOR_BGR2GRAY);


  //中值滤波
  medianBlur(gray, gray, 9);


  //霍夫圆检测
  vector circles;
  HoughCircles(gray, circles, HOUGH_GRADIENT, 1, 50, 100, 100);


  //绘制检测到的圆型
  for (size_t i = 0; i < circles.size(); ++i) {
    Vec3f item = circles[i];
    
    //获取当前圆区域,用黑底
    Mat rectsrc = copyCircleRoi(src, Point(item[0], item[1]), item[2]);
  
    //去高光操作
    Mat rectdst, showmat;
    rectdst = matilluminationChange(rectsrc);


    //极坐标变换
    Mat wpMat1, wpMat2;
    wpMat1 = warpPolarMat(rectsrc);
    wpMat2 = warpPolarMat(rectdst);




    //设置当前圆显示位置
    String title = "rectsrc" + to_string(i);
    String title2 = "rectdst" + to_string(i);
    String title3 = "wpMat1" + to_string(i);
    String title4 = "wpMat2" + to_string(i);


    //间距
    int spacing = 40;
    if ((i % 2) == 0) {
      setshowwindow(rectsrc, title, 0, rectsrc.rows * (i / 2) + spacing * (i / 2));
      setshowwindow(rectdst, title2, rectsrc.rows, rectsrc.rows * (i / 2) + spacing * (i / 2));
      setshowwindow(wpMat1, title3, rectsrc.cols * 2, rectsrc.rows * (i / 2) + spacing * (i / 2));
      setshowwindow(wpMat2, title4, rectsrc.cols * 2, rectsrc.rows * (i / 2) + wpMat1.rows + spacing * (i / 2));
    }
    else {
      setshowwindow(rectsrc, title, (rectsrc.rows * 2 + wpMat1.cols), rectsrc.rows * (i / 2) + (i % 2) + spacing * (i / 2));
      setshowwindow(rectdst, title2, (rectsrc.rows * 2 + wpMat1.cols) + rectdst.rows, rectsrc.rows * (i / 2) + (i % 2) + spacing * (i / 2));
      setshowwindow(wpMat1, title3, (rectsrc.rows * 2 + wpMat1.cols) + rectsrc.cols * 2 , rectsrc.rows * (i / 2) + (i % 2) + spacing * (i / 2));
      setshowwindow(wpMat2, title4, (rectsrc.rows * 2 + wpMat1.cols) + rectsrc.cols * 2, rectsrc.rows * (i / 2) + (i % 2) + wpMat1.rows  + spacing * (i / 2));
    }
    imshow(title, rectsrc);
    imshow(title2, rectdst);
    imshow(title3, wpMat1);
    imshow(title4, wpMat2);
    //break;
  }






  waitKey();


  return 0;
}

b4b0e8a90e6d8c0009ceaa76e2e6c82b.png

8a6d1b8f66bb87a66182601cb92d19bc.png

往期精彩回顾

 

OpenCV消除高亮illuminationChange函数的使用_第11张图片

OpenCV极坐标变换函数warpPolar的使用

 

 

OpenCV消除高亮illuminationChange函数的使用_第12张图片

Android Aidl跨进程通讯(四)--接口回调,服务端向客户端发送数据

 

 

OpenCV消除高亮illuminationChange函数的使用_第13张图片

Android Aidl跨进程通讯(三)--进阶使用

 

你可能感兴趣的:(opencv,人工智能,计算机视觉)