Add methods to export transform as dict for Hydra

src/torchio/transforms/augmentation/composition.py

@@ -81,6 +81,16 @@ def inverse(self, warn: bool = True) -> Compose:
             )
         return result
 
+    def to_hydra_config(self) -> dict:
+        """Return a dictionary representation of the transform for Hydra instantiation."""
+        target = self._get_name_with_module()
+        transform_dict = {'_target_': target}
+        transform_dict['transforms'] = []
+        transform_dict.update(self._get_reproducing_arguments())
+        for transform in self.transforms:
+            transform_dict['transforms'].append(transform.to_hydra_config())
+        return transform_dict
+
 
 class OneOf(RandomTransform):
     """Apply only one of the given transforms.

src/torchio/transforms/transform.py

@@ -596,3 +596,14 @@ def get_mask_from_bounds(
         mask = torch.zeros_like(tensor, dtype=torch.bool)
         mask[:, i0:i1, j0:j1, k0:k1] = True
         return mask
+
+    def _get_name_with_module(self) -> str:
+        """Return the name of the transform including its module."""
+        return f'{self.__class__.__module__}.{self.__class__.__name__}'
+
+    def to_hydra_config(self) -> dict:
+        """Return a dictionary representation of the transform for Hydra instantiation."""
+        target = self._get_name_with_module()
+        transform_dict = {'_target_': target}
+        transform_dict.update(self._get_reproducing_arguments())
+        return transform_dict