def __init__(self, ch=conf.cfmap_ch):
self.ch = ch
self.layers = {}
self.c1, self.c2, self.c3 = G_layers.geo_c1(name="c1"), G_layers.geo_c2(name="c2"), G_layers.geo_c3(name="c3")
self.enc_in_layer = G_layers.input_layer2d(name="enc_in")
self.pretraining_clf(), self.CFmap_generator()
def __init__(self, ch=conf.disc_ch):
self.ch = ch
self.discri_in_layer = G_layers.input_layer2d(name="discriminator_in")
tf.keras.backend.set_image_data_format("channels_last")
self.build_model()