FIXME restore as intended

open-mmlab · Sep 28, 2023 · c20bd14 · c20bd14
1 parent 2890843
commit c20bd14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)