Coherence-Enhancing Shock Filters 代码及详细注释【OpenCV】

本文代码参考自博客

原作者论文下载地址

  

   

    

     

    
    

     

      // CoherenceFilter.cpp : 定义控制台应用程序的入口点。
     
    
   

    

     

    
    

     

      //
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #include "stdafx.h"
     
    
   

    

     

    
    

     

      #include <opencv2/opencv.hpp>
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      using namespace cv;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      /* ==============================================
     
    
   

    

     

    
    

     

      * Coherence-Enhancing Shock Filters
     
    
   

    

     

    
    

     

      * Author:WinCoder@qq.com
     
    
   

    

     

    
    

     

      * inspired by
     
    
   

    

     

    
    

     

      * Joachim Weickert "Coherence-Enhancing Shock Filters"
     
    
   

    

     

    
    

     

      * http://www.mia.uni-saarland.de/Publications/weickert-dagm03.pdf
     
    
   

    

     

    
    

     

      *
     
    
   

    

     

    
    

     

      * Paras:
     
    
   

    

     

    
    

     

      * @img : input image ranging value from 0 to 255.
     
    
   

    

     

    
    

     

      * @sigma : sobel kernel size.
     
    
   

    

     

    
    

     

      * @str_sigma : neighborhood size,see detail in reference[2]
     
    
   

    

     

    
    

     

      * @belnd : blending coefficient.default value 0.5.
     
    
   

    

     

    
    

     

      * @iter : number of iteration.
     
    
   

    

     

    
    

     

      *
     
    
   

    

     

    
    

     

      * Example:
     
    
   

    

     

    
    

     

      * Mat dst = CoherenceFilter(I,11,11,0.5,4);
     
    
   

    

     

    
    

     

      * imshow("shock filter",dst);
     
    
   

    

     

    
    

     

      */
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      Mat CoherenceFilter(Mat img, int sigma, int str_sigma, float blend, int iter)
     
    
   

    

     

    
    

     

      {
     
    
   

    

     

    
    

     

      	Mat I = img.clone();
     
    
   

    

     

    
    

     
	int height = I.rows;
     
    
   

    

     

    
    

     
	int width = I.cols;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
	for (int i = 0; i <iter; i++)
     
    
   

    

     

    
    

     

      	{
     
    
   

    

     

    
    

     

      		Mat gray;
     
    
   

    

     

    
    

     
		cvtColor(I, gray, COLOR_BGR2GRAY);
     
    
   

    

     

    
    

     
		// 计算特征值和特征向量
     
    
   

    

     

    
    

     

      		Mat eigen;
     
    
   

    

     

    
    

     
		cornerEigenValsAndVecs(gray, eigen, str_sigma, 3);
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      		vector<Mat> vec;
     
    
   

    

     

    
    

     
		split(eigen, vec);// vec[0]: λ1, vec[1]: λ2, vec[2]: x1, vec[3]: y1, vec[4]: x2, vec[5]: y2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
		// 主特征向量 w
     
    
   

    

     

    
    

     

      		Mat x, y;
     
    
   

    

     

    
    

     

      		x = vec[2];	// c
     
    
   

    

     

    
    

     

      		y = vec[3];	// s
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
		// Sobel近似求解二阶导数
     
    
   

    

     

    
    

     

      		Mat gxx, gxy, gyy;
     
    
   

    

     

    
    

     
		Sobel(gray, gxx, CV_32F, 2, 0, sigma);
     
    
   

    

     

    
    

     
		Sobel(gray, gxy, CV_32F, 1, 1, sigma);
     
    
   

    

     

    
    

     
		Sobel(gray, gyy, CV_32F, 0, 2, sigma);
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      		Mat ero;
     
    
   

    

     

    
    

     

      		Mat dil;
     
    
   

    

     

    
    

     
		erode(I, ero, Mat());	// 最小值附近进行腐蚀
     
    
   

    

     

    
    

     
		dilate(I, dil, Mat());	// 最大值附近进行膨胀
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      		Mat img1 = ero;
     
    
   

    

     

    
    

     
		for (int nY = 0; nY<height; nY++)
     
    
   

    

     

    
    

     

      		{
     
    
   

    

     

    
    

     
			for (int nX = 0; nX<width; nX++)
     
    
   

    

     

    
    

     

      			{
     
    
   

    

     

    
    

     
				// 边界检测子v(ww)
     
    
   

    

     

    
    

     
				if (x.at<float>(nY, nX)* x.at<float>(nY, nX)* gxx.at<float>(nY, nX)
     
    
   

    

     

    
    

     

      					+ 2 * x.at<float>(nY, nX)* y.at<float>(nY, nX)* gxy.at<float>(nY, nX)
     
    
   

    

     

    
    

     

      					+ y.at<float>(nY, nX)* y.at<float>(nY, nX)* gyy.at<float>(nY, nX)<0)
     
    
   

    

     

    
    

     

      				{
     
    
   

    

     

    
    

     

      					img1.at<Vec3b>(nY, nX) = dil.at<Vec3b>(nY, nX);	// 最大值的影响区域 dil
     
    
   

    

     

    
    

     

      				}
     
    
   

    

     

    
    

     
				// 否则是最小值的影响区域 ero
     
    
   

    

     

    
    

     

      			}
     
    
   

    

     

    
    

     

      		}
     
    
   

    

     

    
    

     
		// 和原图按比例混合
     
    
   

    

     

    
    

     

      		I = I*(1.0 - blend) + img1*blend;
     
    
   

    

     

    
    

     

      	}
     
    
   

    

     

    
    

     
	return I;
     
    
   

    

     

    
    

     

      }
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      int main()
     
    
   

    

     

    
    

     

      {
     
    
   

    

     

    
    

     

      	Mat img = imread("D:/Pictures/beard.jpg", 1);
     
    
   

    

     

    
    

     

      	Mat result = CoherenceFilter(img, 11, 11, 0.5, 2);
     
    
   

    

     

    
    

     
	imshow("Result", result);
     
    
   

    

     

    
    

     
	waitKey(0);
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    return 0;
     
    
   

    

     

    
    

     

      }
     
    
  
 

文章来源: panda1234lee.blog.csdn.net，作者：panda1234lee，版权归原作者所有，如需转载，请联系作者。

原文链接：panda1234lee.blog.csdn.net/article/details/52828214