From cb3ea56a0e18bbaa26ab572b47b78ddb3f7d9575 Mon Sep 17 00:00:00 2001 From: Louis-Dupont <35190946+Louis-Dupont@users.noreply.github.com> Date: Wed, 10 May 2023 16:58:53 +0300 Subject: [PATCH] fix (#972) --- src/super_gradients/training/transforms/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/super_gradients/training/transforms/transforms.py b/src/super_gradients/training/transforms/transforms.py index c024b0a6d2..ba973f4153 100644 --- a/src/super_gradients/training/transforms/transforms.py +++ b/src/super_gradients/training/transforms/transforms.py @@ -850,7 +850,7 @@ def __call__(self, sample: dict) -> dict: return sample def get_equivalent_preprocessing(self) -> List[Dict]: - return [{Processings.DetectionRescale: {"output_shape": self.output_size}}] + return [{Processings.DetectionRescale: {"output_shape": self.output_shape}}] @register_transform(Transforms.DetectionRandomRotate90)