學習OpenCV：濾鏡系列（7）——漩渦

查志強發表於2014-11-25
OpenCV

【原文：http://blog.csdn.net/yangtrees/article/details/9114283】

==============================================

版權所有：小熊不去實驗室CSDN部落格

==============================================


設圖象的漩渦中心為o,設p為影象任意點,只要對p做繞o的旋轉,且旋轉角隨p-o距離的增加而減少,就是"漩渦"變換了.


[cpp] view plaincopyprint?
  1. #include<math.h>  
  2. #include <opencv/cv.h>  
  3. #include <opencv/highgui.h>  
  4.   
  5. using namespace cv;  
  6. using namespace std;  
  7.   
  8. template<typename T> T sqr(T x){return x*x;}  
  9.   
  10. double Pi=3.14;  
  11.   
  12. double Para=20;  
  13.   
  14. int main()  
  15. {  
  16.     Mat src = imread("D:/img/face01.jpg",1);  
  17.     int heigh = src.rows;  
  18.     int width = src.cols;  
  19.     Point center(width/2, heigh/2);  
  20.     Mat img;  
  21.     src.copyTo(img);  
  22.     Mat src1u[3];  
  23.     split(src,src1u);  
  24.   
  25.     for (int y=0; y<heigh; y++)  
  26.     {  
  27.         uchar* imgP = img.ptr<uchar>(y);  
  28.         uchar* srcP = src.ptr<uchar>(y);  
  29.         for(int x=0; x<width; x++)  
  30.         {  
  31.             int R = norm(Point(x,y)-center);  
  32.             double angle = atan2((double)(y-center.y),(double)(x-center.x));  
  33.             double delta=Pi*Para/sqrtf(R+1);  
  34.             int newX = R*cos(angle+delta) + center.x;  
  35.             int newY = R*sin(angle+delta) + center.y;  
  36.   
  37.             if(newX<0) newX=0;  
  38.             if(newX>width-1) newX=width-1;  
  39.             if(newY<0) newY=0;  
  40.             if(newY>heigh-1) newY=heigh-1;  
  41.   
  42.             imgP[3*x] = src1u[0].at<uchar>(newY,newX);  
  43.             imgP[3*x+1] = src1u[1].at<uchar>(newY,newX);  
  44.             imgP[3*x+2] = src1u[2].at<uchar>(newY,newX);  
  45.         }  
  46.     }  
  47.     imshow("vortex",img);  
  48.     waitKey();  
  49.     imwrite("D:/img/漩渦.jpg",img);  
  50. }  

原圖：


漩渦：




相關文章