C++ OpenCV实现图像去阴影

2021-04-21 16:23:04 浏览数 (1)

前言

浏览OpenCV相关文章时看到了《基于OpenCV的图像阴影去除》,源码也是用pytyon实现的,分析了一下其原理,这篇我们用OpenCV的C 版来实现一下。

1

来源

原文链接:https://blog.csdn.net/qq_42722197/article/details/110790222

原文源码:https://github.com/kavyamusty/Shading-removal-of-images

本文用的图片也是源码中提供的测试图片。

#

实现思路

1

图将转为灰度图

2

将灰度图进行膨胀操作

3

膨胀后的图再进行腐蚀操作

4

先膨胀后腐蚀后的图减去原灰度图再取反

5

将取反后的图使用归一化将白色背景修改贴近原图

实现效果

从上图中可以看到,最左边为原图转换的灰度图,中间图为去掉阴影后再归一化后的图,最右边是直接通过去掉阴影实现的图。

代码实现

微卡智享

代码语言:javascript复制
#include <opencv2/opencv.hpp>
#include <iostream>
#include "CvUtils.h"

using namespace cv;
using namespace std;

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

  Mat src = imread("E:/DCIM/Test_image.jpg");
  CvUtils::MatResize(src, 800, 520);
  //imshow("src", src);

  //1.将图像转为灰度图
  Mat gray;
  cvtColor(src, gray, COLOR_BGR2GRAY);
  CvUtils::SetShowWindow(gray, "gray", 0, 30);
  imshow("gray", gray);

  //定义腐蚀和膨胀的结构化元素和迭代次数
  Mat element = getStructuringElement(MORPH_RECT, Size(3, 3));
  int iteration = 9;

  //2.将灰度图进行膨胀操作
  Mat dilateMat;
  morphologyEx(gray, dilateMat, MORPH_DILATE, element, Point(-1, -1), iteration);
  //imshow("dilate", dilateMat);

  //3.将膨胀后的图再进行腐蚀
  Mat erodeMat;
  morphologyEx(dilateMat, erodeMat, MORPH_ERODE, element, cv::Point(-1, -1), iteration);
  //imshow("erode", erodeMat);

  //4.膨胀再腐蚀后的图减去原灰度图再进行取反操作
  Mat calcMat = ~(erodeMat - gray);
  CvUtils::SetShowWindow(calcMat, "calc", gray.cols*2, 30);
  imshow("calc", calcMat);

  //5.使用规一化将原来背景白色的改了和原来灰度图差不多的灰色
  Mat removeShadowMat;
  normalize(calcMat, removeShadowMat, 0, 200, NORM_MINMAX);
  CvUtils::SetShowWindow(removeShadowMat, "dst", 
    gray.cols, 30);
  imshow("dst", removeShadowMat);


  waitKey(0);
  return 0;
}

说明

如果只复制代码可以将CvUtils的引用都去掉,这个类中我写了几个静态函数,主要是加载的图像在当前屏幕分辨率下能够显示全,还有就是自己设定图像显示的区域位置。

源码地址

https://github.com/Vaccae/OpenCVDemoCpp.git

点击下方的原文链接可以跳转到码云的源码地址。

0 人点赞