otthorn/nerf_tutorial
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Save fullrender with the iteration name

Browse source
This commit is contained in:
otthorn 2021-04-30 15:21:55 +02:00
parent 475ef1ad23
commit 6b1829a893
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
main.py
View file

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