augmentation cli options (#467)

* augmentation cli options * pre commit
aramis-lab · Aug 29, 2023 · caea6d0 · caea6d0
1 parent 2ceab5b
commit caea6d0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/clinicadl/utils/cli_param/train_option.py b/clinicadl/utils/cli_param/train_option.py
@@ -169,7 +169,21 @@
 data_augmentation = cli_param.option_group.data_group.option(
     "--data_augmentation",
     "-da",
-    type=click.Choice(["None", "Noise", "Erasing", "CropPad", "Smoothing"]),
+    type=click.Choice(
+        [
+            "None",
+            "Noise",
+            "Erasing",
+            "CropPad",
+            "Smoothing",
+            "Motion",
+            "Ghosting",
+            "Spike",
+            "BiasField",
+            "RandomBlur",
+            "RandomSwap",
+        ]
+    ),
     # default=(),
     multiple=True,
     help="Randomly applies transforms on the training set.",

diff --git a/docs/RandomSearch.md b/docs/RandomSearch.md
@@ -93,7 +93,7 @@ Optional variables:
     - `baseline` (bool) allows to only load `_baseline.tsv` files when set to `True`.
     Sampling function: `choice`. Default: `False`.
     - `data_augmentation` (list of str) is the list of data augmentation transforms applied to the training data.
-    Must be chosen in [`None`, `Noise`, `Erasing`, `CropPad`, `Smoothing`].
+    Must be chosen in[`None`, `Noise`, `Erasing`, `CropPad`, `Smoothing`, `Motion`, `Ghosting`, `Spike`, `BiasField`, `RandomBlur`, `RandomSwap`].
     Sampling function: `fixed`. Default: `False`.
     - `unnormalize` (bool) is a flag to disable min-max normalization that is performed by default.
     Sampling function: `choice`. Default: `False`.