Add method to get image from name (#1368)

fepegar · web-flow · commit fdf02ad73ef3 · 2025-09-02T11:15:58.000+01:00
diff --git a/src/torchio/data/subject.py b/src/torchio/data/subject.py
@@ -360,6 +360,10 @@ def get_images(
         )
         return list(images_dict.values())
 
+    def get_image(self, image_name: str) -> Image:
+        """Get a single image by its name."""
+        return self.get_images_dict(intensity_only=False)[image_name]
+
     def get_first_image(self) -> Image:
         return self.get_images(intensity_only=False)[0]
 

Original file line number	Diff line number	Diff line change
`@@ -360,6 +360,10 @@ def get_images(`
`360`	`360`	`)`
`361`	`361`	`return list(images_dict.values())`
`362`	`362`
	`363`	`+ def get_image(self, image_name: str) -> Image:`
	`364`	`+ """Get a single image by its name."""`
	`365`	`+ return self.get_images_dict(intensity_only=False)[image_name]`
	`366`	`+`
`363`	`367`	`def get_first_image(self) -> Image:`
`364`	`368`	`return self.get_images(intensity_only=False)[0]`
`365`	`369`