def Encoder(self, input_layer,): x = input_layer x = conv(self.keras, x, 256) x = conv(self.keras, x, 512) x = conv(self.keras, x, 768) x = conv(self.keras, x, 1024) x = self.keras.layers.Flatten()(x) return self.keras.models.Model(input_layer, x)
def Encoder(self, input_layer): x = input_layer x = conv(self.keras, x, 128) x = conv(self.keras, x, 256) x = conv(self.keras, x, 512) x = conv(self.keras, x, 1024) x = self.keras.layers.Dense(512)(self.keras.layers.Flatten()(x)) x = self.keras.layers.Dense(8 * 8 * 512)(x) x = self.keras.layers.Reshape((8, 8, 512))(x) x = upscale(self.keras, x, 512) return self.keras.models.Model(input_layer, x)
def Encoder(input_shape): input_layer = self.keras.layers.Input(input_shape) x = input_layer if created_vram_gb >= 4: x = conv(self.keras, x, 128) x = conv(self.keras, x, 256) x = conv(self.keras, x, 512) x = conv(self.keras, x, 1024) x = self.keras.layers.Dense(1024)( self.keras.layers.Flatten()(x)) x = self.keras.layers.Dense(4 * 4 * 1024)(x) x = self.keras.layers.Reshape((4, 4, 1024))(x) x = upscale(self.keras, x, 512) else: x = conv(self.keras, x, 128) x = conv(self.keras, x, 256) x = conv(self.keras, x, 512) x = conv(self.keras, x, 768) x = self.keras.layers.Dense(512)( self.keras.layers.Flatten()(x)) x = self.keras.layers.Dense(4 * 4 * 512)(x) x = self.keras.layers.Reshape((4, 4, 512))(x) x = upscale(self.keras, x, 256) return self.keras.models.Model(input_layer, x)
def Encoder(input_shape): input_layer = self.keras.layers.Input(input_shape) x = input_layer if created_vram_gb >= 5: x = conv(self.keras, x, 128) x = conv(self.keras, x, 256) x = conv(self.keras, x, 512) x = conv(self.keras, x, 1024) x = self.keras.layers.Dense(512)( self.keras.layers.Flatten()(x)) x = self.keras.layers.Dense(8 * 8 * 512)(x) x = self.keras.layers.Reshape((8, 8, 512))(x) x = upscale(self.keras, x, 512) else: x = conv(self.keras, x, 128) x = conv(self.keras, x, 256) x = conv(self.keras, x, 512) x = conv(self.keras, x, 1024) x = self.keras.layers.Dense(256)( self.keras.layers.Flatten()(x)) x = self.keras.layers.Dense(8 * 8 * 256)(x) x = self.keras.layers.Reshape((8, 8, 256))(x) x = upscale(self.keras, x, 256) return self.keras.models.Model(input_layer, x)