Add degrees option to forward model (#397)

Author: eigenvivek

diffdrr/drr.py

@@ -160,14 +160,20 @@ def forward(
     convention: str = None,  # If parameterization is Euler angles, specify convention
     calibration: RigidTransform = None,  # Optional calibration matrix with the detector's intrinsic parameters
     mask_to_channels: bool = False,  # If True, structures from the CT mask are rendered in separate channels
+    degrees: bool = False,  # If parameterization is Euler angles, use degrees instead of radians
     **kwargs,  # Passed to the renderer
 ):
     """Generate DRR with rotational and translational parameters."""
     # Initialize the camera pose
     if parameterization is None:
         pose = args[0]
     else:
-        pose = convert(*args, parameterization=parameterization, convention=convention)
+        pose = convert(
+            *args,
+            parameterization=parameterization,
+            convention=convention,
+            degrees=degrees,
+        )
 
     # Create the source / target points and render the image
     source, target = self.detector(pose, calibration)

notebooks/api/00_drr.ipynb

@@ -279,14 +279,15 @@
     "    convention: str = None,  # If parameterization is Euler angles, specify convention\n",
     "    calibration: RigidTransform = None,  # Optional calibration matrix with the detector's intrinsic parameters\n",
     "    mask_to_channels: bool = False,  # If True, structures from the CT mask are rendered in separate channels\n",
+    "    degrees: bool = False,  # If parameterization is Euler angles, use degrees instead of radians\n",
     "    **kwargs,  # Passed to the renderer\n",
     "):\n",
     "    \"\"\"Generate DRR with rotational and translational parameters.\"\"\"\n",
     "    # Initialize the camera pose\n",
     "    if parameterization is None:\n",
     "        pose = args[0]\n",
     "    else:\n",
-    "        pose = convert(*args, parameterization=parameterization, convention=convention)\n",
+    "        pose = convert(*args, parameterization=parameterization, convention=convention, degrees=degrees)\n",
     "\n",
     "    # Create the source / target points and render the image\n",
     "    source, target = self.detector(pose, calibration)\n",