gan处理自己的数据集_pytorch:实现简单的GAN示例(MNIST数据集)

# -*- coding: utf-8 -*-

"""

Created on Sat Oct 13 10:22:45 2018

@author: www

"""

import torch

from torch import nn

from torch.autograd import Variable

import torchvision.transforms as tfs

from torch.utils.data import DataLoader, sampler

from torchvision.datasets import MNIST

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.gridspec as gridspec

plt.rcParams['figure.figsize'] = (10.0, 8.0) # 设置画图的尺寸

plt.rcParams['image.interpolation'] = 'nearest'

plt.rcParams['image.cmap'] = 'gray'

def show_images(images): # 定义画图工具

images = np.reshape(images, [images.shape[0], -1])

sqrtn = int(np.ceil(np.sqrt(images.shape[0])))

sqrtimg = int(np.ceil(np.sqrt(images.shape[1])))

<