python实现对图片尺寸进行resize的完整代码,包含对一个目录下的所有图片的读取,resize,和重新存储:
import torch.utils.data as data from glob import glob from PIL import Image import torchvision.transforms as transforms import argparse import os import imageio class DataSet(data.Dataset): def __init__(self, img_dir, resize): super(DataSet, self).__init__() self.img_paths = glob('{:s}/*'.format(img_dir)) self.transform = transforms.Compose([transforms.Resize(size=(resize, resize))]) def __getitem__(self, item): img = Image.open(self.img_paths[item]).convert('RGB') img = self.transform(img) return img, self.img_paths[item] def __len__(self): return len(self.img_paths) if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('--img_dir', type=str, default='') parser.add_argument('--resize', type=int, default=256) parser.add_argument('--save_dir', type=str, default='') args = parser.parse_args() if not os.path.exists(args.save_dir): os.mkdir(args.save_dir) dataset = DataSet(args.img_dir, args.resize) print('dataset:', len(dataset)) for i in range(len(dataset)): img, path = dataset[i] path = os.path.basename(path) print('Processing:', path) imageio.imwrite(args.save_dir+'/{:s}'.format(path), img)