(#3) Defense: Fix rank of images

betarixm · May 2, 2022 · 788e901 · 788e901
1 parent afb63f0
commit 788e901
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/defense/models.py b/src/defense/models.py
@@ -161,8 +161,11 @@ def custom_callbacks(self) -> List[keras.callbacks.Callback]:
     with defense_model.tensorboard_file_writer().as_default():
         for idx, (x, _) in enumerate(test_set):
             y = defense_model.predict(x)
+
+            tf.summary.image(f"(Defense) Original images", x, step=idx)
+
             tf.summary.image(
-                f"{args.defense.upper()} processing result", [x, y], step=idx
+                f"(Defense) {args.defense.upper()} processing result", y, step=idx
             )
 
             progress.add(1)