diff --git a/mmyolo/datasets/transforms/transforms.py b/mmyolo/datasets/transforms/transforms.py index 3a9b8a0a5..275e33bba 100644 --- a/mmyolo/datasets/transforms/transforms.py +++ b/mmyolo/datasets/transforms/transforms.py @@ -264,7 +264,7 @@ def _resize_img(self, results: dict): results['pad_param_origin'] = results['pad_param'] * \ np.repeat(ratio, 2) - if 'gt_masks' in results: + if self.half_pad_param or 'gt_masks' in results: results['pad_param'] = np.array( [padding_h / 2, padding_h / 2, padding_w / 2, padding_w / 2], dtype=np.float32)