dcgan_mnist.py 文件源码

def __init__(self):
        self.img_rows = 28
        self.img_cols = 28
        self.channel = 1

        #(XX_train, YY_train),(X_test, Y_test) = mnist.load_data()
        print "111111"

        trainData, trainLabels = loadData('./mnisttrain',1000)
        self.x_train = trainData
        #self.x_train = XX_train
        #self.x_train = input_data.read_data_sets("mnist",\
        #   one_hot=True).train.images
        print "222222"
        self.x_train = self.x_train.reshape(-1, self.img_rows,\
            self.img_cols, 1).astype(np.float32)
        print "333333"


        self.DCGAN = DCGAN()
        self.discriminator =  self.DCGAN.discriminator_model()
        self.adversarial = self.DCGAN.adversarial_model()
        self.generator = self.DCGAN.generator()