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

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

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

1. MATLAB代码实现
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) > 1
    im = 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函数实现
matlab 复制代码
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. 辅助函数
matlab 复制代码
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);
end

function im = shave(im, border)
    % 删除边框
    im = im(border(1)+1:end-border(1), border(2)+1:end-border(2));
end

function psnr = compute_psnr(im1, im2)
    % 计算PSNR
    mse = mean((im1(:) - im2(:)).^2);
    psnr = 10 * log10(255^2 / mse);
end

参考代码 MATLAB 实现 SRCNN 图像超分辨率重建 youwenfan.com/contentcsb/50734.html

代码可以在MATLAB中实现基于SRCNN的图像超分辨率重建。该代码包括图像预处理、双三次插值、SRCNN重建以及结果评估等步骤。

相关推荐
一个会的不多的人13 分钟前
C# NX二次开发:面收集器控件和曲线收集器控件详解
开发语言·c#
Freak嵌入式25 分钟前
一文速通 Python 并行计算:教程总结
开发语言·python
2401_8370885040 分钟前
setup 语法糖核心要点
开发语言·前端·javascript
我爱学嵌入式2 小时前
C语言:第18天笔记
c语言·开发语言·笔记
楚Y6同学3 小时前
QT之QSS的使用方法和常用控件的样式设置
开发语言·qt
Yolo566Q4 小时前
“R语言+遥感”的水环境综合评价方法实践技术应用
开发语言·r语言
文火冰糖的硅基工坊6 小时前
[激光原理与应用-317]:光学设计 - Solidworks - 草图
开发语言·python·信息可视化·系统架构
草莓熊Lotso6 小时前
【C语言强化训练16天】--从基础到进阶的蜕变之旅:Day10
c语言·开发语言·经验分享·算法·强化
草明6 小时前
docker stats 增加一列容器名称的显示
java·开发语言·docker
He1955016 小时前
Go初级二
开发语言·后端·golang