From 6b1829a89345d747523e095de3a90545a829a788 Mon Sep 17 00:00:00 2001
From: Solal Nathan <otthorn@crans.org>
Date: Fri, 30 Apr 2021 15:21:55 +0200
Subject: [PATCH] Save fullrender with the iteration name

---
 main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 8c1acf2..97c124e 100644
--- a/main.py
+++ b/main.py
@@ -424,6 +424,7 @@ def show_full_render(
     target_silhouette,
     loss_history_color,
     loss_history_sil,
+    iteration
 ):
     """
     This is a helper function for visualizing the intermediate results of the
@@ -481,7 +482,7 @@ def show_full_render(
             ax_.axis("off")
         ax_.set_title(title_)
     fig.canvas.draw()
-    fig.savefig("results/fullrender.png")
+    fig.savefig(f"results/fullrender_{iteration:05d}.png")
     #fig.show()
     #display.clear_output(wait=True)
     #display.display(fig)
@@ -624,6 +625,7 @@ for iteration in range(n_iter):
             target_silhouettes[show_idx][0],
             loss_history_color,
             loss_history_sil,
+            iteration,
         )
 
 ###############################################################################