训练样本切分小图

GTX_AI

训练样本切分小图

(1)这个【train文件夹】，含有000.png, 001.png, 002.png, ……, 999.png；以及对应的标签000_mask.png, 001_mask.png, 002_mask.png, ……, 999_mask.png. 图像大小为512x512；
(2)目的：第一：将文件夹train，将图像依次切分为256x256大小，得到【train2文件夹】；
第二：将得到的256x256大小图像的train2文件夹，合并为512x512大小的图像，得到【train3文件夹】；

GTX_AI

第一：切分图像：

1. clc,clear, close all
   
2. warning off
   
3. FilePath='../data/train';
   
4. fileList = dir(FilePath);
   
5. savePath='../data/train2';
   
6. delete([savePath, '/', '*.**']);
   
7. for i=1:length(fileList)
   
8.     if strcmp(fileList(i).name,'.')==1||strcmp(fileList(i).name,'..')==1
   
9.         continue;
   
10.     else
    
11.         fileList(i).name;
    
12.         img = imread([FilePath,'/',fileList(i).name]);
    
13.         % Crop Parts
    
14.         CropImagesPartsFun(img, fileList(i).name, 256, 256, 150, savePath);
    
15.     end
    
16. end
    
17. disp('切分完成！！！');
    

复制代码

相应地的切分函数为：

1. function CropImagesPartsFun(img, imageName, newW, newH, OverlapRate, savePath)
   
2. %         imageName=fileList(i).name;
   
3. %         newW=256;
   
4. %         newH=256;
   
5. %         OverlapRate=200;
   
6. [w,h,d]=size(img);
   
7. img1 = zeros(newW,newH,d);
   
8. k=0;
   
9. for i=1:OverlapRate:w-1
   
10.     for j=1:OverlapRate:h-1
    
11.         iW=i+newW-1;
    
12.         jH=j+newH-1;
    
13.         if(iW>w)
    
14.             iW=w;
    
15.         end
    
16.         if(jH>h)
    
17.             jH=h;
    
18.         end
    
19.         img1 = img(iW-newW+1:iW,jH-newH+1:jH,d);
    
20.         k=k+1;
    
21.         % .png图像
    
22.         if(length(imageName)>=8)
    
23.             if(isequal(imageName(end-7:end),'mask.png'))
    
24.                 imageNewName = [imageName(1:end-8), num2str(k) , '_mask.png'];
    
25.                 imwrite(img1, [savePath,'/',imageNewName])
    
26.             elseif(isequal(imageName(end-3:end),'.png'))
    
27.                 imageNewName = [imageName(1:end-4), num2str(k) , '.png'];
    
28.                 imwrite(img1, [savePath,'/',imageNewName])
    
29.             end
    
30.         elseif(length(imageName)>4 && length(imageName)<8)
    
31.             imageNewName = [imageName(1:end-4),'_', num2str(k) , '.png'];
    
32.             imwrite(img1, [savePath,'/',imageNewName])
    
33.         end
    
34.         
    
35.     end
    
36. end

复制代码

GTX_AI

第二：将切分的小图合并为原始大图：
主函数如下：

1. clc,clear, close all
   
2. warning off
   
3. FilePath='../data/train2';
   
4. fileList = dir(FilePath);
   
5. savePath='../data/train3';
   
6. delete([savePath, '/', '*.**']);
   
7. % 提取唯一出现的名称ID
   
8. fileList2=[];
   
9. for i=1:length(fileList)
   
10.     if strcmp(fileList(i).name,'.')==1||strcmp(fileList(i).name,'..')==1
    
11.         continue;
    
12.     else
    
13.         imageName=fileList(i).name;
    
14.         if(isequal(imageName(end-5:end),'_1.png'))
    
15.             fileList2 = [fileList2; imageName(1:end-6)];
    
16.         end
    
17.     end
    
18. end
    
19. % 根据图像ID名称，结合切分的小图，进行大图合图
    
20. for i=1:length(fileList2)
    
21.     imageName=fileList2(i,:);
    
22.     imagePathName = [FilePath, '/', imageName];
    
23.     % Merge Parts
    
24.     newW=256;          % 小图宽
    
25.     newH=256;          % 小图高
    
26.     OverlapRate=150;   % 大图切分小图时，重叠大小
    
27.     w=512;             % 大图宽
    
28.     h=512;             % 大图高
    
29.     MergeImagesPartsFun( FilePath, imageName, newW, newH, OverlapRate, w, h, savePath);
    
30. end
    
31. disp('合图完成！！！');

复制代码

MergeImagesPartsFun函数如下：

1. function MergeImagesPartsFun( FilePath, imageName, newW,newH, OverlapRate, w,h, savePath)
   
2. %         imageName=fileList(i).name; 大图唯一ID名称
   
3. %         newW=256; 小图宽
   
4. %         newH=256; 小图高
   
5. %         OverlapRate=200; 大图切分小图所用的重叠大小
   
6. % knums:  小图数量
   
7. % w: 大图宽
   
8. % h: 大图高
   
9. k=0;
   
10. for i=1:OverlapRate:w-1
    
11.     for j=1:OverlapRate:h-1
    
12.         iW=i+newW-1;
    
13.         jH=j+newH-1;
    
14.         if(iW>w)
    
15.             iW=w;
    
16.         end
    
17.         if(jH>h)
    
18.             jH=h;
    
19.         end
    
20.         % .png图像
    
21.         k=k+1;
    
22.         imageNewName = [FilePath, '/', imageName,'_', num2str(k) , '.png'];
    
23.         img1 = imread( imageNewName );
    
24.         img(iW-newW+1:iW,jH-newH+1:jH, :) = img1;
    
25.     end
    
26. end
    
27. % 保存图像
    
28. imwrite(img, [savePath,'/', [imageName, '.png']])
    
29. clear im1 img

复制代码