Allow user to pass custom intrinsic matrix

eigenvivek · Mar 1, 2024 · fb75e13 · fb75e13
1 parent 3ae7b1b
commit fb75e13
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 23 deletions.
diff --git a/diffpose/deepfluoro.py b/diffpose/deepfluoro.py
@@ -184,11 +184,11 @@ def convert_diffdrr_to_deepfluoro(specimen, pose: RigidTransform):
 
 
 class Evaluator:
-    def __init__(self, specimen, idx):
+    def __init__(self, specimen, idx, intrinsic=None):
         # Save matrices to device
         self.translate = specimen.translate
         self.flip_xz = specimen.flip_xz
-        self.intrinsic = specimen.intrinsic
+        self.intrinsic = specimen.intrinsic if intrinsic is None else intrinsic
         self.intrinsic_inv = specimen.intrinsic.inverse()
 
         # Get gt fiducial locations

diff --git a/notebooks/api/00_deepfluoro.ipynb b/notebooks/api/00_deepfluoro.ipynb