diff --git a/tests/test_datasets/test_transforms/test_common_transforms.py b/tests/test_datasets/test_transforms/test_common_transforms.py index 2818081dca..ac221c2af5 100644 --- a/tests/test_datasets/test_transforms/test_common_transforms.py +++ b/tests/test_datasets/test_transforms/test_common_transforms.py @@ -49,7 +49,7 @@ def test_transform(self): results.update(bbox_center=center, bbox_scale=scale) results = transform(results) self.assertTrue(np.allclose(results['bbox_center'], center)) - self.assertTrue(np.allclose(results['bbox_scale'], scale)) + self.assertTrue(np.allclose(results['bbox_scale'], scale * padding)) def test_repr(self): transform = GetBBoxCenterScale(padding=1.25)