Initialize calibration matrix with (#304)

eigenvivek · Jul 10, 2024 · 9af7c19 · 9af7c19
1 parent 24a85e5
commit 9af7c19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/diffdrr/drr.py b/diffdrr/drr.py
@@ -65,7 +65,6 @@ def __init__(
 
         # Initialize the volume and world geometry
         self.subject = subject
-        self.volume = subject.volume.data.squeeze()
         self.register_buffer(
             "_affine",
             torch.as_tensor(subject.volume.affine, dtype=torch.float32).unsqueeze(0),
@@ -197,7 +196,7 @@ def set_intrinsics(
         n_subsample=self.detector.n_subsample,
         reverse_x_axis=self.detector.reverse_x_axis,
         reorient=self.subject.reorient,
-    ).to(self.volume)
+    ).to(self.density)
 
 # %% ../notebooks/api/00_drr.ipynb 12
 @patch

diff --git a/notebooks/api/00_drr.ipynb b/notebooks/api/00_drr.ipynb
@@ -160,7 +160,6 @@
     "\n",
     "        # Initialize the volume and world geometry\n",
     "        self.subject = subject\n",
-    "        self.volume = subject.volume.data.squeeze()\n",
     "        self.register_buffer(\n",
     "            \"_affine\",\n",
     "            torch.as_tensor(subject.volume.affine, dtype=torch.float32).unsqueeze(0),\n",
@@ -324,7 +323,7 @@
     "        n_subsample=self.detector.n_subsample,\n",
     "        reverse_x_axis=self.detector.reverse_x_axis,\n",
     "        reorient=self.subject.reorient,\n",
-    "    ).to(self.volume)"
+    "    ).to(self.density)"
    ]
   },
   {