本文整理汇总了Python中datasets.ImageDataset方法的典型用法代码示例。如果您正苦于以下问题:Python datasets.ImageDataset方法的具体用法?Python datasets.ImageDataset怎么用?Python datasets.ImageDataset使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类datasets
的用法示例。
在下文中一共展示了datasets.ImageDataset方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: main
# 需要导入模块: import datasets [as 别名]
# 或者: from datasets import ImageDataset [as 别名]
def main():
device = torch.device("cuda:0" if FLAGS.cuda else "cpu")
if FLAGS.train:
print('Loading data...\n')
transform = [transforms.Resize(int(FLAGS.img_size*1.12), Image.BICUBIC),
transforms.RandomCrop((FLAGS.img_size, FLAGS.img_size)),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))]
dataloader = DataLoader(ImageDataset(os.path.join(FLAGS.data_dir, FLAGS.dataset),
transform=transform, unaligned=True, mode='train'),
batch_size=FLAGS.batch_size, shuffle=True, num_workers=2)
test_dataloader = DataLoader(ImageDataset(os.path.join(FLAGS.data_dir, FLAGS.dataset),
transform=transform, unaligned=True, mode='test'),
batch_size=FLAGS.test_batch_size, shuffle=True, num_workers=2)
print('Creating model...\n')
model = Model(FLAGS.model, device, dataloader, test_dataloader, FLAGS.channels, FLAGS.img_size, FLAGS.num_blocks)
model.create_optim(FLAGS.lr)
# Train
model.train(FLAGS.epochs, FLAGS.log_interval, FLAGS.out_dir, True)
model.save_to('')
else:
model = Model(FLAGS.model, device, None, test_dataloader, FLAGS.channels, FLAGS.img_size, FLAGS.num_blocks)
model.load_from(FLAGS.out_dir)
model.eval(mode=1, batch_size=FLAGS.batch_size)
开发者ID:PacktPublishing,项目名称:Hands-On-Generative-Adversarial-Networks-with-PyTorch-1.x,代码行数:31,代码来源:main.py