【李宏毅2022 机器学习春】hw6_GAN（不懂..）

数据集查看

数据集下载需要root权限，我先下载到有root权限的服务器，再scp传输。阿里云往外传 限流了…

（好用的库-model && dataset）Hugging Face ： https://huggingface.co/datasets/LeoFeng/MLHW_6

temp_dataset = get_dataset(os.path.join(workspace_dir, 'faces'))

# transforms.ToPILImage(),
images = [temp_dataset[i] for i in range(4)]
grid_img = torchvision.utils.make_grid(images, nrow=4)
plt.figure(figsize=(10,10))
plt.imshow(grid_img.permute(1, 2, 0))
plt.show()