基于偏振物理模型的水下图像去雾MATLAB实现

一、核心算法框架

1. 偏振成像物理模型

水下偏振去雾基于以下数学模型：

I=J⋅e−βd+B⋅(1−e−βd)I=J⋅e^{−βd}+B⋅(1−e^{−βd})I=J⋅e−βd+B⋅(1−e−βd)

其中：

III：观测图像强度
JJJ：目标反射光强度
BBB：后向散射光强度
βββ：衰减系数
ddd：传输距离

通过正交偏振片获取两幅图像：

I0°=[J+B]⋅e−βd/2I_{0°}=[J+B]⋅e^{−βd/2}I0°=[J+B]⋅e−βd/2
I90°=[J−B]⋅e−βd/2I_{90°}=[J−B]⋅e^{−βd/2}I90°=[J−B]⋅e−βd/2

2. 关键参数估计

透射率估计：
t=1−12(I0°−I90°)t=1−\frac{1}{2}(I_{0°}−I_{90°})t=1−21(I0°−I90°)
背景光估计：
B=max(I0°,I90°)e−βdB=\frac{max(I_{0°},I_{90°})}{e^{−βd}}B=e−βdmax(I0°,I90°)

二、MATLAB实现代码

1. 数据预处理

% 读取正交偏振图像img0=imread('p0.jpg');img90=imread('p90.jpg');% 转换为灰度图像gray0=rgb2gray(img0);gray90=rgb2gray(img90);% 计算偏振度DoP=(img0-img90)./(img0+img90+eps);

2. 背景光估计

% 选择背景区域（手动选择ROI）figure,imshow(img0),title('选择背景区域');roi=drawrectangle('Label','Background');mask=createMask(roi);% 计算背景光偏振态B_p=mean2(DoP(mask));B_intensity=mean2(img0(mask));

3. 透射率计算

% 修正系数计算w=0.95;% 根据实验调整t=1-w*(1-2*(img0-img90)./(img0+img90+eps));% 透射率约束t(t<0.1)=0.1;t(t>0.9)=0.9;

4. 图像恢复

% 估计衰减系数beta=-log(t)./d;% d为预设传输距离% 恢复清晰图像J=(img0-B_intensity)./exp(-beta)+B_intensity;J=im2uint8(J);

5. 后处理优化

% 直方图均衡化J_eq=adapthisteq(J);% 色彩校正J_lab=rgb2lab(J_eq);J_lab(:,:,1)=lab2uint8(adapthisteq(lab2uint8(J_lab(:,:,1))));J_final=lab2rgb(J_lab);

三、关键参数优化

1. 传输距离估计

% 基于场景深度估计depth_map=stereoBM(imread('left.png'),50,25);beta=beta.*(1+0.1*depth_map);% 线性深度补偿

2. 偏振噪声抑制

% 偏振差分滤波DoP_filt=imgaussfilt(DoP,1.5);DoP_filt(DoP_filt<0)=0;DoP_filt(DoP_filt>1)=1;

四、完整代码实现

functiondehazed_img=underwater_polarization_dehaze(img0,img90,d)% 输入：% img0: 0°偏振图像% img90: 90°偏振图像% d: 预设传输距离(m)% 预处理gray0=rgb2gray(img0);gray90=rgb2gray(img90);DoP=(double(img0)-double(img90))./(double(img0)+double(img90)+eps);% 背景光估计figure,imshow(img0),title('选择背景区域');roi=drawrectangle('Label','Background');mask=createMask(roi);B_intensity=mean2(img0(mask));% 透射率计算w=0.95;% 实验优化参数t=1-w*(1-2*(img0-img90)./(img0+img90+eps));t=max(min(t,0.9),0.1);% 图像恢复beta=-log(t)./d;J=(double(img0)-B_intensity)./exp(-beta)+B_intensity;% 后处理J=im2uint8(J);J_eq=adapthisteq(J);J_lab=rgb2lab(J_eq);J_lab(:,:,1)=lab2uint8(adapthisteq(lab2uint8(J_lab(:,:,1))));dehazed_img=lab2rgb(J_lab);end