Argument size now is optional

Gradiant · Aug 21, 2024 · 434021b · 434021b
1 parent e2241da
commit 434021b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmcls/datasets/pipelines/transforms.py b/mmcls/datasets/pipelines/transforms.py
@@ -693,7 +693,7 @@ class Resize(object):
     """
 
     def __init__(self,
-                 size,
+                 size=None,
                  interpolation='bilinear',
                  adaptive_side='short',
                  backend='cv2',
@@ -715,6 +715,7 @@ def __init__(self,
 
         self.adaptive_side = adaptive_side
         self.adaptive_resize = False
+        self.size = img_scale
         if isinstance(size, int):
             assert size > 0
             size = (size, size)