GTX_AI 发表于 2019-7-26 21:59:36

读取文件夹内图像--方法1

读取文件夹内图像:
图像以数字命名:
参考:【训练样本切分小图】
供深度学习算法调用:Pytorch_Unet图像分割
import torch.utils.data as data
import PIL.Image as Image
import os

def make_dataset(root):
    imgs=[]
    n=len(os.listdir(root))//2
#    for i in range(100):
    for i in range(100):
      for j in range(16):
            img=os.path.join(root,"%03d_%d.png"%(i,j+1))
            mask=os.path.join(root,"%03d_%d_mask.png"%(i,j+1))
            imgs.append((img,mask))
    return imgs

def make_dataset512(root):
    imgs=[]
    n=len(os.listdir(root))//10
    for i in range(n):
      img=os.path.join(root,"%03d.png"%i)
      mask=os.path.join(root,"%03d_mask.png"%i)
      imgs.append((img,mask))
    return imgs


class LiverDataset(data.Dataset):
    def __init__(self, root, transform=None, target_transform=None):
      imgs = make_dataset(root)
      self.imgs = imgs
      self.transform = transform
      self.target_transform = target_transform

    def __getitem__(self, index):
      x_path, y_path = self.imgs
      img_x = Image.open(x_path)
      img_y = Image.open(y_path)
      if self.transform is not None:
            img_x = self.transform(img_x)
      if self.target_transform is not None:
            img_y = self.target_transform(img_y)
      return img_x, img_y

    def __len__(self):
      return len(self.imgs)











页: [1]
查看完整版本: 读取文件夹内图像--方法1