From 1680ac74cb0c5e20c82f36f4580f20c3e06716d5 Mon Sep 17 00:00:00 2001
From: Nic Ma <nma@nvidia.com>
Date: Mon, 20 Dec 2021 12:25:12 +0800
Subject: [PATCH] [DLMED] add 6 new transform images

Signed-off-by: Nic Ma <nma@nvidia.com>
---
 docs/source/transforms.rst                    | 12 +++++++
 monai/transforms/smooth_field/array.py        |  3 --
 .../transforms/utils_create_transform_ims.py  | 32 +++++++++++++++++++
 3 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/docs/source/transforms.rst b/docs/source/transforms.rst
index 653647a3fa..4885fd33bf 100644
--- a/docs/source/transforms.rst
+++ b/docs/source/transforms.rst
@@ -647,6 +647,8 @@ Spatial
 
 `GridDistortion`
 """"""""""""""""
+.. image:: https://github.com/Project-MONAI/DocImages/raw/main/transforms/GridDistortion.png
+    :alt: example of GridDistortion
 .. autoclass:: GridDistortion
     :members:
     :special-members: __call__
@@ -735,12 +737,16 @@ Smooth Field
 
 `RandSmoothFieldAdjustContrast`
 """""""""""""""""""""""""""""""
+.. image:: https://github.com/Project-MONAI/DocImages/raw/main/transforms/RandSmoothFieldAdjustContrast.png
+    :alt: example of RandSmoothFieldAdjustContrast
 .. autoclass:: RandSmoothFieldAdjustContrast
     :members:
     :special-members: __call__
 
 `RandSmoothFieldAdjustIntensity`
 """"""""""""""""""""""""""""""""
+.. image:: https://github.com/Project-MONAI/DocImages/raw/main/transforms/RandSmoothFieldAdjustIntensity.png
+    :alt: example of RandSmoothFieldAdjustIntensity
 .. autoclass:: RandSmoothFieldAdjustIntensity
     :members:
     :special-members: __call__
@@ -1521,6 +1527,8 @@ Spatial (Dict)
 
 `GridDistortiond`
 """""""""""""""""
+.. image:: https://github.com/Project-MONAI/DocImages/raw/main/transforms/GridDistortiond.png
+    :alt: example of GridDistortiond
 .. autoclass:: GridDistortiond
     :members:
     :special-members: __call__
@@ -1538,12 +1546,16 @@ Smooth Field (Dict)
 
 `RandSmoothFieldAdjustContrastd`
 """"""""""""""""""""""""""""""""
+.. image:: https://github.com/Project-MONAI/DocImages/raw/main/transforms/RandSmoothFieldAdjustContrastd.png
+    :alt: example of RandSmoothFieldAdjustContrastd
 .. autoclass:: RandSmoothFieldAdjustContrastd
     :members:
     :special-members: __call__
 
 `RandSmoothFieldAdjustIntensityd`
 """""""""""""""""""""""""""""""""
+.. image:: https://github.com/Project-MONAI/DocImages/raw/main/transforms/RandSmoothFieldAdjustIntensityd.png
+    :alt: example of RandSmoothFieldAdjustIntensityd
 .. autoclass:: RandSmoothFieldAdjustIntensityd
     :members:
     :special-members: __call__
diff --git a/monai/transforms/smooth_field/array.py b/monai/transforms/smooth_field/array.py
index b8016dd3fd..714bcf738e 100644
--- a/monai/transforms/smooth_field/array.py
+++ b/monai/transforms/smooth_field/array.py
@@ -214,9 +214,6 @@ def __call__(self, img: np.ndarray, randomize: bool = True):
         if not self._do_transform:
             return img
 
-        img_min = img.min()
-        img_max = img.max()
-
         field = self.sfield()
         rfield, *_ = convert_to_dst_type(field, img)
 
diff --git a/monai/transforms/utils_create_transform_ims.py b/monai/transforms/utils_create_transform_ims.py
index e7939b764f..ab4e952f90 100644
--- a/monai/transforms/utils_create_transform_ims.py
+++ b/monai/transforms/utils_create_transform_ims.py
@@ -145,7 +145,10 @@
 )
 from monai.transforms.post.array import KeepLargestConnectedComponent, LabelFilter, LabelToContour
 from monai.transforms.post.dictionary import AsDiscreted, KeepLargestConnectedComponentd, LabelFilterd, LabelToContourd
+from monai.transforms.smooth_field.array import RandSmoothFieldAdjustContrast, RandSmoothFieldAdjustIntensity
+from monai.transforms.smooth_field.dictionary import RandSmoothFieldAdjustContrastd, RandSmoothFieldAdjustIntensityd
 from monai.transforms.spatial.array import (
+    GridDistortion,
     Rand2DElastic,
     RandAffine,
     RandAxisFlip,
@@ -155,6 +158,7 @@
     Spacing,
 )
 from monai.transforms.spatial.dictionary import (
+    GridDistortiond,
     Rand2DElasticd,
     RandAffined,
     RandAxisFlipd,
@@ -668,9 +672,37 @@ def create_transform_im(
     create_transform_im(
         KeepLargestConnectedComponentd, dict(keys=CommonKeys.LABEL, applied_labels=1), data_binary, is_post=True, ndim=2
     )
+    create_transform_im(
+        GridDistortion, dict(num_cells=3, distort_steps=[(1.5,) * 4] * 3, mode="nearest", padding_mode="zeros"), data
+    )
+    create_transform_im(
+        GridDistortiond,
+        dict(
+            keys=keys, num_cells=3, distort_steps=[(1.5,) * 4] * 3, mode=["bilinear", "nearest"], padding_mode="zeros"
+        ),
+        data,
+    )
     create_transform_im(RandGridDistortion, dict(num_cells=3, prob=1.0, distort_limit=(-0.1, 0.1)), data)
     create_transform_im(
         RandGridDistortiond,
         dict(keys=keys, num_cells=4, prob=1.0, distort_limit=(-0.2, 0.2), mode=["bilinear", "nearest"]),
         data,
     )
+    create_transform_im(
+        RandSmoothFieldAdjustContrast, dict(spatial_size=(217, 217, 217), rand_size=(100, 100, 100), prob=1.0), data
+    )
+    create_transform_im(
+        RandSmoothFieldAdjustContrastd,
+        dict(keys=keys, spatial_size=(217, 217, 217), rand_size=(100, 100, 100), prob=1.0),
+        data,
+    )
+    create_transform_im(
+        RandSmoothFieldAdjustIntensity,
+        dict(spatial_size=(217, 217, 217), rand_size=(100, 100, 100), prob=1.0, gamma=(0.5, 4.5)),
+        data,
+    )
+    create_transform_im(
+        RandSmoothFieldAdjustIntensityd,
+        dict(keys=keys, spatial_size=(217, 217, 217), rand_size=(100, 100, 100), prob=1.0, gamma=(0.5, 4.5)),
+        data,
+    )