MATLAB中基于CNN实现图像超分辨率重建

原创于 2025-12-18 09:40:33 发布 · 321 阅读

CC 4.0 BY-SA版权

文章标签：

在MATLAB中基于CNN实现图像超分辨率重建，主要有使用预训练模型和从零训练自定义网络两种路径。

特性	方案一：使用预训练VDSR模型（推荐初学者）	方案二：训练自定义网络（适合深入研究）
核心优点	快速、简单，无需训练，立即可用	灵活，可针对特定数据或缩放因子优化
主要步骤	加载模型 → 预处理图像 → 网络预测 → 后处理	数据准备 → 网络设计 → 模型训练 → 评估应用
所需时间	几分钟	几小时至数天（取决于数据和硬件）
关键函数/工具	`load`, `activations`, `imresize`	`imageDatastore`, `randomPatchExtractionDatastore`, `trainNetwork`

方案一：快速上手（使用预训练VDSR模型）

这是最快捷的方法。MATLAB提供了在大型数据集上预训练好的VDSR网络，能直接用于2倍、3倍、4倍等常见倍率的超分。

% 1. 加载预训练的VDSR网络模型
load('trainedVDSR-Epoch-100-ScaleFactors-234.mat'); % 确保此.mat文件在MATLAB路径中

% 2. 读取并预处理低分辨率图像
lowResImg = imread('your_low_resolution_image.jpg');
lowResImg = im2double(lowResImg); % 转换为双精度

% 转换为YCbCr颜色空间，VDSR仅处理亮度通道(Y)以提升效率
ycbcrImg = rgb2ycbcr(lowResImg);
yChannel = ycbcrImg(:, :, 1); % 亮度通道
cbChannel = ycbcrImg(:, :, 2); % 色度通道Cb
crChannel = ycbcrImg(:, :, 3); % 色度通道Cr

% 3. 使用双三次插值将亮度通道放大到目标尺寸（作为网络输入的基础）
scaleFactor = 3; % 例如，放大3倍
targetSize = size(yChannel) * scaleFactor;
yBicubic = imresize(yChannel, targetSize, 'bicubic');

% 4. 使用VDSR网络预测残差图像（高频细节）
residual = activations(net, yBicubic, 41); % '41'是VDSR的输出层名称或索引
residual = double(residual); % 确保数据类型

% 5. 重建高分辨率亮度通道：基础图像 + 残差（高频细节）
yHighRes = yBicubic + residual;

% 6. 合并通道并转回RGB
% 色度通道仅使用双三次插值放大
cbHighRes = imresize(cbChannel, targetSize, 'bicubic');
crHighRes = imresize(crChannel, targetSize, 'bicubic');
% 合并三个通道
highResYcbcr = cat(3, yHighRes, cbHighRes, crHighRes);
highResRgb = ycbcr2rgb(highResYcbcr);

% 7. 显示与比较结果
figure;
subplot(1,2,1); imshow(lowResImg); title('原始低分辨率图像');
subplot(1,2,2); imshow(highResRgb); title(['VDSR超分辨率重建 (x', num2str(scaleFactor), ')']);

方案二：深入定制（从零开始训练网络）

如果你有特定数据集或研究需求，可以训练自己的网络。以训练一个VDSR网络为例，主要步骤如下：

准备训练数据：需要高分辨率（HR）图像数据集。程序会自动生成对应的低分辨率（LR）图像对。
构建网络架构：使用MATLAB的Deep Learning Toolbox逐层搭建VDSR等CNN网络。
配置并启动训练：设置优化器、学习率、迭代次数等参数进行训练。
评估与应用：使用训练好的模型对新图像进行超分，并使用PSNR、SSIM等指标客观评价。

关键代码结构概览：

% 1. 准备训练数据（示例流程）
trainImagesDir = 'path_to_high_resolution_images';
% 使用 imageDatastore 管理图像
pristineImages = imageDatastore(trainImagesDir, 'FileExtensions', '.jpg');
% 调用辅助函数生成LR-HR训练对
scaleFactors = [2 3 4];
createVDSRTrainingSet(pristineImages, scaleFactors, upsampledDir, residualDir);

% 2. 构建VDSR网络（示例：20个卷积层）
layers = [
    imageInputLayer([41 41 1], 'Name', 'input') % 输入41x41的图像块
    convolution2dLayer(3, 64, 'Padding', 1, 'Name', 'conv1')
    reluLayer('Name', 'relu1')
    % ... 重复18组卷积层+ReLU层 ...
    convolution2dLayer(3, 1, 'Padding', 1, 'Name', 'conv20') % 输出残差图像
    regressionLayer('Name', 'output') % 回归任务层
];
lgraph = layerGraph(layers);

% 3. 配置训练选项
options = trainingOptions('adam', ...
    'InitialLearnRate', 0.001, ...
    'MaxEpochs', 100, ...
    'MiniBatchSize', 64, ...
    'Plots', 'training-progress');

% 4. 开始训练（需要大量时间和计算资源，建议使用GPU）
net = trainNetwork(trainingData, lgraph, options);