From f994d449f1e0b58bca686f910c908ce22e48403a Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 25 Oct 2022 21:32:17 +0100 Subject: [PATCH] Layer2Image: fix --- aimodel/src/lib/ai/components/LayerStack2Image.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/aimodel/src/lib/ai/components/LayerStack2Image.py b/aimodel/src/lib/ai/components/LayerStack2Image.py index 4b9d7f9..23745f2 100644 --- a/aimodel/src/lib/ai/components/LayerStack2Image.py +++ b/aimodel/src/lib/ai/components/LayerStack2Image.py @@ -19,8 +19,7 @@ class LayerStack2Image(tf.keras.layers.Layer): return config def call(self, input_thing, **kwargs): - result = tf.stack([ input_thing for i in range(self.param_target_width) ], axis=-1) - result = tf.stack([ result for i in range(self.param_target_height) ], axis=-1) - result = tf.stack([ result ], axis=-1) # channel dimension + result = tf.stack([ input_thing for i in range(self.param_target_width) ], axis=-2) + result = tf.stack([ result for i in range(self.param_target_height) ], axis=-2) return result \ No newline at end of file