图像数据增广

addpath('test');
addpath('train');
fid = fopen('C:\Users\Byte\Desktop\data.txt','wt')
file_path = 'C:\Users\Byte\Desktop\test_label\';   %图像文件夹路径
save_path = 'C:\Users\Byte\Desktop\cc_test\';   %图像文件夹路径

% img_path_list = dir(strcat(file_path,'*.jpg')); %获取文件夹中所有jpg格式图像
img_path_list = dir(strcat(file_path,'*.png'));

img_num = length(img_path_list);  %获取总数
if img_num > 0
    for j = 1 : img_num
        img_name = img_path_list(j).name;
        image = imread(strcat(file_path,img_name));
        img_name_core = img_name(1:end-4);    %除去后缀名
        
        fprintf('%d %d %s\n',i,j,strcat(file_path,img_name));
        fprintf('%s\n',strcat(file_path,img_name(1:end-4)));
        %图像增广
        
    
        %翻转flipdim
        img_u = flipdim(image,1);
        img_r = flipdim(image,2);
        img_ur = flipdim(img_r,1);
        imwrite(img_u,strcat(save_path,img_name_core,'_u.png'));
        imwrite(img_r,strcat(save_path,img_name_core,'_r.png'));
        imwrite(img_ur,strcat(save_path,img_name_core,'_ur.png'));     
        
        %亮度调节
        
        %img_bright = imadjust(image,[0,1],[0,1],0.5);
        img_dark = imadjust(image,[0,1],[0,1],1.5);
        %imwrite(img_bright,strcat(save_path,img_name_core,'_b.jpg'));
        imwrite(img_dark,strcat(save_path,img_name_core,'_d.png'));
        
        %img_ub = imadjust(img_u,[0,1],[0,1],0.5);
        img_ud = imadjust(img_u,[0,1],[0,1],1.5);
        %imwrite(img_ub,strcat(save_path,img_name_core,'_ub.jpg'));
        imwrite(img_ud,strcat(save_path,img_name_core,'_ud.png'));
        
        %img_rb = imadjust(img_r,[0,1],[0,1],0.5);
        img_rd = imadjust(img_r,[0,1],[0,1],1.5);
        %imwrite(img_rb,strcat(save_path,img_name_core,'_rb.jpg'));
        imwrite(img_rd,strcat(save_path,img_name_core,'_rd.png'));
        
        %img_urb = imadjust(img_ur,[0,1],[0,1],0.5);
        img_urd = imadjust(img_ur,[0,1],[0,1],1.5);
        %imwrite(img_urb,strcat(save_path,img_name_core,'_urb.jpg'));
        imwrite(img_urd,strcat(save_path,img_name_core,'_urd.png'));
        
        
        
        
        %旋转
        for k = 1:5
            img_u = imrotate(img_u,72,'bilinear','loose');
            img_u = imcrop(img_u,[67 67 511 511]);
            imwrite(img_u,strcat(save_path,img_name_core,'_',num2str(k),'_u.png'));
            img_ur = imrotate(img_ur,72,'bilinear','loose');
            img_ur = imcrop(img_ur,[67 67 511 511]);
            imwrite(img_ur,strcat(save_path,img_name_core,'_',num2str(k),'_ur.png'));
            img_r = imrotate(img_r,72,'bilinear','loose');
            img_r = imcrop(img_r,[67 67 511 511]);
            imwrite(img_r,strcat(save_path,img_name_core,'_',num2str(k),'_r.png'));
            image = imrotate(image,72,'bilinear','loose');
            image = imcrop(image,[67 67 511 511]);
            imwrite(image,strcat(save_path,img_name_core,'_',num2str(k),'.png'));
        end
        
    end
end