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, ) ###############################################################################