making ready for skirts

2022-09-30 21:53:17 +00:00
parent 6a2f9ff4ff
commit dca5368704
22 changed files with 700 additions and 219 deletions
@@ -75,9 +75,6 @@ class UpscalerBasic(tf.keras.Model):
    def call(self, inputs, training=None):
        img, mask = inputs

-        if img.shape[1] == 3:
-            img = tf.transpose(img, [0,2,3,1])
-
        x = self.flatten(img)
        x = self.seq(x, training=training)
        x = self.reshape(x)