clc
clear
SOURCE_DIR='G:/watecrop';
fid=fopen('water_train.txt','w');
fid2=fopen('water_val.txt','w');
Path=dir(SOURCE_DIR);
for i=3:length(Path)
sub_dir = Path(i).name;
pi = fullfile(SOURCE_DIR, sub_dir);
local_lujing=dir(pi);
label=i-3;
u1=randperm(length(local_lujing)-2)+2;
val_name=local_lujing(u1(1)).name;
aaa=fullfile(sub_dir,val_name)
fprintf(fid2,'%s',aaa);
fprintf(fid2,' ');
fprintf(fid2, num2str(label));
fprintf(fid2,'\n');
for j=2:length(u1)
img_name = local_lujing(u1(j)).name;
bbb=fullfile(sub_dir,img_name)
fprintf(fid,'%s',bbb);
fprintf(fid,' ');
fprintf(fid, num2str(label));
fprintf(fid,'\n');
end
end
fclose(fid);
fclose(fid2);
这里需要说明的是:matlab在linux下和windows下