【图像增强】基于matlab单尺度Retinex图像增强【含Matlab源码 3961期】.zip

clc clear close all % 输入原始图像 image = imread('ant.jpg'); % 构造高斯滤波器 [N1, M1, dim] = size(image); sigma = 250; F = fspecial('gaussian', [N1,M1], sigma); Efft = fft2(double(F)); %%%%%%%%%%%%%%%%%%% 获取R通道值,并进行对数处理 %%%%%%%%%%%%%%%%%%%%%%% R = image(:, :, 1); R0 = double(R); Rlog = log(R0+1); Rfft2 = fft2(R0); DR0 = Rfft2.* Efft; DR = ifft2(DR0); % 求反射分量，并归一化处理 DRlog = log(DR +1); Rr = Rlog - DRlog; EXPRr = exp(Rr); MIN = min(min(EXPRr)); MAX = max(max(EXPRr)); EXPRr = (EXPRr - MIN)/(MAX - MIN); % 调用CLAHE对比度增强函数 EXPRr = adapthisteq(EXPRr); %%%%%%%%%%%%%%%%%%% 获取G通道值,并进行对数处理 %%%%%%%%%%%%%%%%%%%%%%% G = image(:, :, 2); G0 = double(G); Glog = log(G0+1); Gfft2 = fft2(G0); DG0 = Gfft2.* Efft; DG = ifft2(DG0); % 求反射分量，并归一化处理 DGlog = log(DG +1); Gg = Glog - DGlog; EXPGg = exp(Gg); MIN = min(min(EXPGg)); MAX = max(max(EXPGg)); EXPGg = (EXPGg - MIN)/(MAX - MIN); % 调用CLAHE对比度增强函数 EXPGg = adapthisteq(EXPGg); %%%%%%%%%%%%%%%%%%% 获取B通道值,并进行对数处理 %%%%%%%%%%%%%%%%%%%%%%% B = image(:, :, 3); B0 = double(B); Blog = log(B0+1); Bfft2 = fft2(B0); DB0 = Bfft2.* Efft; DB = ifft2(DB0); % 求反射分量，并归一化处理 DBlog = log(DB+1); Bb = Blog - DBlog; EXPBb = exp(Bb); MIN = min(min(EXPBb)); MAX = max(max(EXPBb)); EXPBb = (EXPBb - MIN)/(MAX - MIN); % 调用CLAHE对比度增强函数 EXPBb = adapthisteq(EXPBb); % 合成SSR彩色图像 ssr_rgb = cat(3, EXPRr, EXPGg, EXPBb)*255; %%%%%%%%%%%%%%%%%%%%%%%%%%%%% 计算增强图像的评价指标 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% improved_gray_image = rgb2gray(uint8(ssr_rgb)); gray_image = rgb2gray(image); % Mean [M_out,N_in] = Mean( improved_gray_image,gray_image ); % MSE k = 3; mse1 = MSE(gray_image,k); mse2 = MSE(improved_gray_image,k); % PSNR psnr = PSNR(improved_gray_image,gray_image); % information entropy，IE ie_src = IE(gray_image,k); ie = IE(improved_gray_image,k); % SSIM ssim = SSIM( improved_gray_image,gray_image ); index = [M_out,N_in,mse1,mse2,psnr,ie_src,ie,ssim]; %******************************* 图像输出显示 ******************************% figure(1) imshow(image,[]);title('原始图像'); figure(2) imshow(uint8(ssr_rgb));title('SSR增强的彩色图像'); % 显示增强后的图像直方图 figure(3) imhist(improved_gray_image);title('增强图像直方图'); % xlabel('增强图像')