draw the background when I do generation

spacetx · dalessioluca · Nov 10, 2020 · Nov 11, 2020 · Nov 11, 2020 · Nov 11, 2020
commit 22ee2d1b751d26f766c84f337e6e04898b2c69ff
diff --git a/src/main_mnist.py b/src/main_mnist.py
@@ -238,7 +238,8 @@
                     #log_concordance(concordance=concordance_vs_gt, prefix="concordance_vs_gt_")
 
                     generated: Output = vae.generate(imgs_in=reference_imgs,
-                                                     draw_boxes=True)
+                                                     draw_boxes=True,
+                                                     draw_bg=True)
                     plot_generation(generated, epoch=epoch, prefix="gen_", experiment=exp)
 
                     test_loss = test_metrics.loss