鱼眼镜头的distortion校正【matlab】

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%         作者：WWC        %%%%%%%%

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%%  功能：畸变矫正
clc;
clear;
close all;
%% 读取图像
A=imread('D:\文件及下载相关\图片\distortion2.jpg');
[m,n,k]=size(A);
R=n/2;
w=3*R;
h=3*R;
xo=m/2;
yo=n/2;
f=2*R/pi;
for u=1:wfor v=1:hho=sqrt((u-xo)^2+(v-yo)^2)+1;h1=f*atan2(ho,f);x=h1*(u-xo)/ho+xo;y=h1*(v-yo)/ho+yo;x=round(x);y=round(y);if(x>2*R || y>2*R || x<1 || y<1)continue;endC(u,v,1)=A(x,y,1);C(u,v,2)=A(x,y,2);C(u,v,3)=A(x,y,3);end
end
C=uint8(C);
imshow(C);

鱼眼镜头畸变校正。（直接处理彩色图）

代码存在错误：貌似只能处理宽比长大的，竖着的。和方的。