LLNet模型實現——訓練數據準備之Matlab圖像格式轉換

1. 背景

主要實現將LLNet模型論文中的PGM格式的圖像數據進行處理轉換

2. 代碼實現

function converTool(fromDir, fromFormat, toDir, toFormat, toPrefix)
% Inputs:
%       fromDir: 輸入的文件目錄
%       fromFormat: 輸入的文件格式
%       toDir: 保存文件的目錄,層數同fromDir
%       toFormat: 保存文件的格式
%       toPrefix: 保存文件的前綴
% Author: HSW
% Date: 2018-05-05

filePaths = searchRoot(fromDir, fromFormat);
if ~exist(toDir, 'dir')
    mkdir(toDir)
end
dirLen = length(fromDir);

fileCnt = length(filePaths);
for idx = 1 : fileCnt
    filePath = filePaths{idx};
    idxs = strfind(filePath, '\');
    idxs2 = strfind(filePath, '.');
    fileName = filePath(idxs(end) + 1 : idxs2(end) - 1);
    from_img = imread(filePath);
    saveName = [fileName, toFormat];
    if ~isempty(toPrefix)
        saveName = [toPrefix, saveName];
    end
    saveDir = fullfile(toDir, filePath(dirLen + 1 : idxs(end)));
    if ~exist(saveDir, 'dir')
        mkdir(saveDir)
    end
    savePath = [saveDir, saveName];
    % 根據需要進行增加即可
    if ~isempty(strfind(toFormat, 'jpg'))
        imwrite(from_img, savePath, 'jpg');
    elseif ~isempty(strfind(toFormat, 'bmp'))
        imwrite(from_img, savePath, 'bmp');
    elseif ~isempty(strfind(savePath, 'png'))
        imwrite(from_img, savePath, 'png');
    end
end

end


3. 代碼效果(左邊爲PGM格式數據集合,右邊爲轉換獲取的PNG格式數據, 根據PGM格式數據的目錄結構,建立相同的文件結構)

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章