OpenCV实现图像呈现球面凹陷

代码如下，图片路径按需求修改

#include <iostream>
#include <string>
#include <opencv2/opencv.hpp>
#include <cv.h>
#include <highgui.h>
#include <cxcore.hpp>
#include <math.h>
#include <time.h>

#define pi 3.1415926

using namespace std;
using namespace cv;
 
// define the show image
void Show_Image(Mat& Image, const string& str)
{
    namedWindow(str.c_str(),CV_WINDOW_AUTOSIZE);
    imshow(str.c_str(), Image);
}

int main()
{
    string Img_name("cat.jpeg");
    Mat Img;
    Img=imread(Img_name);
 
    Mat Img_out(Img.size(), CV_8UC3);
    Img.copyTo(Img_out);
 
    int width=Img.cols;
    int height=Img.rows;
 
    float e;
    float a,b, a0, b0, a1,b1;
    float alpha=1.0;
    float K=pi/2;
 
    a=width/2; b=height/2;
    e=(float)width/(float)height;
 
    Point2i Center(width/2, height/2);
 
    float new_x, new_y;
    float p,q,x1,y1,x0,y0;
    float theta;
 
    for (int y=0; y<height; y++)
    {
        for (int x=0; x<width; x++)
        {
            y0=Center.y-y;
            x0=x-Center.x;
            theta=atan(y0*e/(x0+0.0001));
            if(x0<0)   theta=theta+pi;
            a0=x0/cos(theta);
            b0=y0/sin(theta+0.0001);
            if(a0>a || b0>b)  continue;
 
            a1=a*sin(a0/a*K);
            b1=b*sin(b0/b*K);
 
            a1=(a1-a0)*(alpha)+a0;
            b1=(b1-b0)*(alpha)+b0;
 
            new_x=a1*cos(theta);
            new_y=b1*sin(theta);
 
            new_x=Center.x+new_x;
            new_y=Center.y-new_y;
 
            if(new_x<0)         new_x=0;
            if(new_x>=width-1)  new_x=width-2;
            if(new_y<0)         new_y=0;
            if(new_y>=height-1) new_y=height-2;
 
            x1=(int)new_x;
            y1=(int)new_y;
 
            p=new_x-x1;
            q=new_y-y1;
 
            for (int k=0; k<3; k++)
            {
                Img_out.at<Vec3b>(y, x)[k]=(1-p)*(1-q)*Img.at<Vec3b>(y1, x1)[k]+
                                            (p)*(1-q)*Img.at<Vec3b>(y1,x1+1)[k]+
                                            (1-p)*(q)*Img.at<Vec3b>(y1+1,x1)[k]+
                                            (p)*(q)*Img.at<Vec3b>(y1+1,x1+1)[k];
            }
        }
    }
 
	Show_Image(Img_out, "out");
	imwrite("Out.jpg", Img_out);
 
	waitKey();
}

原图：

效果图：