pytorch中数据预处理基本操作

import torch
from torchvision.utils import save_image
from torchvision.datasets import ImageFolder
from torchvision import transforms
from torch.utils.data import DataLoader
import numpy
import os
img_path = './hh/'
out_path = './0_real_train/'
os.makedirs('./0_real_train', exist_ok=True)

tf = transforms.Compose([
    transforms.Resize(128),
    transforms.ToTensor()
])
dataset = ImageFolder(img_path, transform=tf)
dataloader = DataLoader(dataset, batch_size=1, shuffle=False)
i = 0
for img, _ in dataloader:
    i += 1
    img = img.cuda()
    save_image(img, out_path+f"{i}.png", nrow=1, normalize=False)