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

来源:互联网 发布:php个人自动发卡源码 编辑:程序博客网 时间:2024/06/07 07:03

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

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    endend
0 0
原创粉丝点击