Add metric to animation

eigenvivek · eigenvivek · commit 1f860097ebd2 · 2023-09-05T09:28:35.000-04:00
diff --git a/diffdrr/visualization.py b/diffdrr/visualization.py
@@ -107,7 +107,7 @@ def make_fig(ground_truth):
             )
             itr = drr(rotations, translations)
             _ = plot_drr(itr, axs=ax_opt)
-            ax_opt.set(xlabel="Moving DRR")
+            ax_opt.set(xlabel=f"Moving DRR (loss = {row['loss']:.3f})")
             fig.savefig(f"{tmpdir}/{idx}.png")
             plt.close(fig)
             idxs.append(idx)
diff --git a/notebooks/api/04_visualization.ipynb b/notebooks/api/04_visualization.ipynb
@@ -154,7 +154,7 @@
     "            translations = torch.tensor(row[[\"bx\", \"by\", \"bz\"]].values).unsqueeze(0).to(device)\n",
     "            itr = drr(rotations, translations)\n",
     "            _ = plot_drr(itr, axs=ax_opt)\n",
-    "            ax_opt.set(xlabel=\"Moving DRR\")\n",
+    "            ax_opt.set(xlabel=f\"Moving DRR (loss = {row['loss']:.3f})\")\n",
     "            fig.savefig(f\"{tmpdir}/{idx}.png\")\n",
     "            plt.close(fig)\n",
     "            idxs.append(idx)\n",
diff --git a/notebooks/tutorials/optimizers.ipynb b/notebooks/tutorials/optimizers.ipynb

Original file line number	Diff line number	Diff line change
`@@ -107,7 +107,7 @@ def make_fig(ground_truth):`
`107`	`107`	`)`
`108`	`108`	`itr = drr(rotations, translations)`
`109`	`109`	`_ = plot_drr(itr, axs=ax_opt)`
`110`		`- ax_opt.set(xlabel="Moving DRR")`
	`110`	`+ ax_opt.set(xlabel=f"Moving DRR (loss = {row['loss']:.3f})")`
`111`	`111`	`fig.savefig(f"{tmpdir}/{idx}.png")`
`112`	`112`	`plt.close(fig)`
`113`	`113`	`idxs.append(idx)`