DeepLearning: 数据预处理3:分割数据集(matlab代码)

组合使用以下两个函数就行了,结合前面两篇内容。

randperm(m);
copyfile(a,b)

data路径下已经存在test和train两个文件夹,想要从trian文件夹中拿出25%用于验证,剩下的作为训练。验证的样本被移动到val文件夹。

clear;close all ;clc

%data文件夹下面放置test和train两个文件夹
dir_path = '/home/li/lilai/myMatlabcode/code1/data';
% 获取dir_path下的文件(夹)

dir_list = dir(dir_path);
dir_list_num = length(dir_list);
%第一层循环:
for i = 3:dir_list_num

    if strcmp(dir_list(i).name , 'train')
        dir_sub1_list = dir([dir_path,'/',dir_list(i).name]);
        dir_sub1_list_num = length(dir_sub1_list);
        %第二层循环:
        for j = 3:dir_sub1_list_num
            dir_sub2_list = dir([dir_path,'/',dir_list(i).name,'/',dir_sub1_list(j).name]);
            dir_sub2_list_num = length(dir_sub2_list);
            position_split = round((dir_sub2_list_num-2)/4);
            list_move = randperm(dir_sub2_list_num-2)+2;
            dir_path_new = [dir_path,'/','val','/',dir_sub1_list(j).name];
            mkdir(dir_path_new);

            %第三层循环:
            for k = 1:position_split
                img_oldfile = [dir_path,'/',dir_list(i).name,'/',dir_sub1_list(j).name,'/',dir_sub2_list(list_move(k)).name];

                img_newfile = [dir_path_new];

                copyfile(img_oldfile,img_newfile);

            end
        end
    end

end

你可能感兴趣的:(深度学习)