超分双三次退化数据集制作

python和matlab生成数据集的方法有所差异,公认的方法为matlab生成的数据集,以下matlab代码取自RDN。

function Prepare_TrainData_HR_LR_BI()
%% settings
path_save = 'F:\shujuji\hongwai\SR1280';
path_src = 'F:\shujuji\hongwai\SR1280\SR1280train';
ext               =  {'*.jpg','*.png','*.bmp'};
filepaths           =  [];
for i = 1 : length(ext)
    filepaths = cat(1,filepaths, dir(fullfile(path_src, ext{i})));
end
nb_im = length(filepaths);
SR1280train = [];

for idx_im = 1:nb_im
    fprintf('Read HR :%d\n', idx_im);
    ImHR = imread(fullfile(path_src, filepaths(idx_im).name));
    SR1280train{idx_im} = ImHR;
end
%% generate and save LR via imresize() with Bicubic

for IdxIm = 1:nb_im
    fprintf('IdxIm=%d\n', IdxIm);
    ImHR = SR1280train{IdxIm};
    
    ImLRx2 = imresize(ImHR, 1/2, 'bicubic');
    ImLRx3 = imresize(ImHR, 1/3, 'bicubic');
    ImLRx4 = imresize(ImHR, 1/4, 'bicubic');
    % name image
    digit = IdxIm;
    fileName = num2str(IdxIm);
    while digit < 10000
        fileName = ['0', fileName];
        digit = digit*10;
    end

    FolderLRx2 = fullfile(path_save, 'SR1280train_LR_bicubic', 'X2');
    FolderLRx3 = fullfile(path_save, 'SR1280train_LR_bicubic', 'X3');
    FolderLRx4 = fullfile(path_save, 'SR1280train_LR_bicubic', 'X4');

    if ~exist(FolderLRx2)
        mkdir(FolderLRx2)
    end
    if ~exist(FolderLRx3)
        mkdir(FolderLRx3)
    end
    if ~exist(FolderLRx4)
        mkdir(FolderLRx4)
    end

    NameLRx2 = fullfile(FolderLRx2, [fileName, 'x2.jpg']);
    NameLRx3 = fullfile(FolderLRx3, [fileName, 'x3.jpg']);
    NameLRx4 = fullfile(FolderLRx4, [fileName, 'x4.jpg']);
    % save image
 
    imwrite(ImLRx2, NameLRx2, 'jpg');
    imwrite(ImLRx3, NameLRx3, 'jpg');
    imwrite(ImLRx4, NameLRx4, 'jpg');
end


end
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值