基于MATLAB实现传统图像去噪算法-中值滤波算法高分项目.zip

clear,clc; pauseTime = 1; data_path = "..\Set12"; ext = ["*.jpg", "*.png", "*.jpeg"]; filePaths = []; for i = 1 : length(ext) filePaths = cat(1,filePaths, dir(fullfile(data_path,ext(i)))); end noise_leval = [10,15,20,25,30,35,40,45,50,55,60,65,70]; for ii = 1:length(noise_leval) PSNRs = zeros(1, length(filePaths)); SSIMs = zeros(1, length(filePaths)); sigma = noise_leval(ii); for jj = 1:length(filePaths) % 原图像 originImage = im2double(imread(filePaths(jj).name)); % 添加高斯噪声 imageWithNoise = single(originImage + sigma/255*randn(size(originImage))); [rows, cols] = size(originImage); y = imageWithNoise; % 中值滤波算法 % 指定模板尺寸 boxSize = 3; template = zeros(boxSize); for i = 1:rows-boxSize+1 for j = 1:cols-boxSize+1 % 取模板内像素 template = imageWithNoise(i:i+(boxSize-1),j:j+(boxSize-1)); % 用中值替换模板中心点像素值 m = median(template(:)); y(i+(boxSize-1)/2,j+(boxSize-1)/2) = m; end end % 计算psnr和ssim PSNRs(jj) = psnr(im2uint8(originImage), im2uint8(y)); SSIMs(jj) = ssim(im2uint8(originImage), im2uint8(y)); imshow(cat(2,im2uint8(originImage),im2uint8(imageWithNoise),im2uint8(y))); title(['sigma=',num2str(sigma),' ',filePaths(jj).name,' psnr=',num2str(PSNRs(jj),'%2.2f'),'dB',' ssim=',num2str(SSIMs(jj),'%2.4f')]) drawnow; pause(pauseTime) end disp(["sigma:",sigma,"psnr:",mean(PSNRs),"ssim:", mean(SSIMs)]); end