Add missing paper reference

Fixes #254.
fepegar · Aug 15, 2020 · c83d9c8 · c83d9c8
1 parent 38a944f
commit c83d9c8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/torchio/transforms/augmentation/intensity/random_labels_to_image.py b/torchio/transforms/augmentation/intensity/random_labels_to_image.py
@@ -12,8 +12,10 @@
 class RandomLabelsToImage(RandomTransform, IntensityTransform):
     r"""Generate an image from a segmentation.
 
-    Based on the work by `Billot et al., A Learning Strategy for
-    Contrast-agnostic MRI Segmentation <https://arxiv.org/abs/2003.01995>`_.
+    Based on the works by Billot et al.: `A Learning Strategy for
+    Contrast-agnostic MRI Segmentation <https://arxiv.org/abs/2003.01995>`_
+    and `Partial Volume Segmentation of Brain MRI Scans of any Resolution and
+    Contrast <https://arxiv.org/abs/2004.10221>`.
 
     Args:
         label_key: String designating the label map in the sample