[Fix] Fix the issue open-mmlab#1711 "GaussianBlur doesn't work" (open…

…-mmlab#1722) * Fix issue 1711. GaussianBlur. * Fix UT --------- Co-authored-by: mzr1996 <[email protected]>
ASHORE1225 · Jul 25, 2023 · 2b8d8ee · 2b8d8ee
1 parent 60d780f
commit 2b8d8ee
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/mmpretrain/datasets/transforms/auto_augment.py b/mmpretrain/datasets/transforms/auto_augment.py
@@ -1169,7 +1169,7 @@ def transform(self, results):
 
         img = results['img']
         pil_img = Image.fromarray(img)
-        pil_img.filter(ImageFilter.GaussianBlur(radius=radius))
+        pil_img = pil_img.filter(ImageFilter.GaussianBlur(radius=radius))
         results['img'] = np.array(pil_img, dtype=img.dtype)
 
         return results

diff --git a/tests/test_datasets/test_transforms/test_auto_augment.py b/tests/test_datasets/test_transforms/test_auto_augment.py
@@ -1285,9 +1285,10 @@ def test_initialize(self):
 
     def test_transform(self):
         transform_func = 'PIL.ImageFilter.GaussianBlur'
+        from PIL.ImageFilter import GaussianBlur
 
         # test params inputs
-        with patch(transform_func, autospec=True) as mock:
+        with patch(transform_func, wraps=GaussianBlur) as mock:
             cfg = {
                 **self.DEFAULT_ARGS,
                 'radius': 0.5,
@@ -1297,7 +1298,7 @@ def test_transform(self):
             mock.assert_called_once_with(radius=0.5)
 
         # test prob
-        with patch(transform_func, autospec=True) as mock:
+        with patch(transform_func, wraps=GaussianBlur) as mock:
             cfg = {
                 **self.DEFAULT_ARGS,
                 'radius': 0.5,
@@ -1307,7 +1308,7 @@ def test_transform(self):
             mock.assert_not_called()
 
         # test magnitude_range
-        with patch(transform_func, autospec=True) as mock:
+        with patch(transform_func, wraps=GaussianBlur) as mock:
             cfg = {
                 **self.DEFAULT_ARGS,
                 'magnitude_range': (0.1, 2),