当前位置：首页 > news >正文

MATLAB 实现 SRCNN 图像超分辨率重建

news 2025/8/2 21:12:08

MATLAB 实现 SRCNN 图像超分辨率重建

MATLAB代码实现，用于基于三层卷积神经网络的图像超分辨率重建。代码参考了多个来源，结合了SRCNN的典型实现步骤。

1. MATLAB代码实现

% 超分辨率卷积神经网络(SRCNN)的测试代码
% 参考文献：Chao Dong, Chen Change Loy, Kaiming He, Xiaoou Tang.
% 学习深度卷积网络的图像超分辨率，欧洲计算机视觉会议记录，2014年% 清理环境
close all;
clear all;% 读取真实图像
im = imread('Set5\butterfly_GT.bmp'); % 替换为你的图像路径
% im = imread('Set14\zebra.bmp');% 设定参数
up_scale = 3; % 放大倍数
model = 'model\x3.mat'; % 替换为你的模型路径% 仅在照度方面工作
if size(im, 3) > 1im = rgb2ycbcr(im);im = im(:, :, 1);
end
im_gnd = modcrop(im, up_scale); % 裁剪图像以匹配放大倍数
im_gnd = double(im_gnd) / 255; % 归一化到[0, 1]% 双三次插值
im_l = imresize(im_gnd, 1 / up_scale, 'bicubic'); % 缩小
im_b = imresize(im_l, up_scale, 'bicubic'); % 再放大% SRCNN重建
im_h = SRCNN(model, im_b);% 删除边框
im_h = shave(uint8(im_h * 255), [up_scale, up_scale]);
im_gnd = shave(uint8(im_gnd * 255), [up_scale, up_scale]);
im_b = shave(uint8(im_b * 255), [up_scale, up_scale]);% 计算PSNR
psnr_bic = compute_psnr(im_gnd, im_b);
psnr_srcnn = compute_psnr(im_gnd, im_h);% 显示结果
fprintf('双三次插值的峰值信噪比: %f dB\n', psnr_bic);
fprintf('SRCNN重建的峰值信噪比: %f dB\n', psnr_srcnn);figure, imshow(im_b); title('双三次插值');
figure, imshow(im_h); title('SRCNN重建');% 保存结果
imwrite(im_b, ['双三次插值' '.bmp']);
imwrite(im_h, ['SRCNN重建' '.bmp']);

2. SRCNN函数实现

function im_h = SRCNN(model, im_b)% 加载预训练模型load(model);% 提取特征F1 = conv2(im_b, W1, 'same') + B1;F1 = max(0, F1); % ReLU激活函数% 非线性映射F2 = conv2(F1, W2, 'same') + B2;F2 = max(0, F2); % ReLU激活函数% 重建im_h = conv2(F2, W3, 'same') + B3;
end

3. 辅助函数

function im = modcrop(im, scale)% 裁剪图像以匹配放大倍数[h, w] = size(im);h = h - mod(h, scale);w = w - mod(w, scale);im = im(1:h, 1:w);
endfunction im = shave(im, border)% 删除边框im = im(border(1)+1:end-border(1), border(2)+1:end-border(2));
endfunction psnr = compute_psnr(im1, im2)% 计算PSNRmse = mean((im1(:) - im2(:)).^2);psnr = 10 * log10(255^2 / mse);
end