读取文件夹内图像--方法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]