Generator.py 文件源码

def __init__(self, 
                    g_size=(3, 128, 64),
                    g_nb_filters=128,
                    g_nb_coding=200,
                    g_init=None,
                **kwargs):
        super(MLP, self).__init__(**kwargs)

        self.g_size = g_size
        self.g_nb_filters = g_nb_filters
        self.g_nb_coding = g_nb_coding 
        self.g_init = g_init if g_init is not None else InitNormal()

        c, h, w = g_size # h and w should be multiply of 16
        nf = g_nb_filters

        self.add(Dense(g_nb_filters, input_shape=(g_nb_coding,)) )
        self.add(Activation('relu'))
        self.add(Dense(g_nb_filters))
        self.add(Activation('relu'))
        self.add(Dense(g_nb_filters, ))
        self.add(Activation('relu'))
        self.add(Dense(np.prod(g_size), ))
        self.add(Reshape(g_size))
        g_init(self)