Add metric to animation

eigenvivek · Sep 5, 2023 · 1f86009 · 1f86009
1 parent e94e1db
commit 1f86009
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 28 deletions.
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