fix typo in RandomFlip.__repr__

2022-06-27 18:36:14 +08:00 · 2022-06-27 18:36:14 +08:00 · 08bc449888
parent 1a57a76be5
commit 08bc449888
1 changed files with 2 additions and 2 deletions
--- a/mmcv/transforms/processing.py
+++ b/mmcv/transforms/processing.py
@ -1197,8 +1197,8 @@ class RandomFlip(BaseTransform):

    def __repr__(self) -> str:
        repr_str = self.__class__.__name__
-        repr_str += f'(scale={self.prob}, '
-        repr_str += f'interpolation={self.direction})'
+        repr_str += f'(prob={self.prob}, '
+        repr_str += f'direction={self.direction})'

        return repr_str