Fix numpy warning (#1632)

Deci-AI · Nov 13, 2023 · aa58407 · aa58407
1 parent 3067278
commit aa58407
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/super_gradients/training/transforms/utils.py b/src/super_gradients/training/transforms/utils.py
@@ -133,6 +133,9 @@ def _pad_image(image: np.ndarray, padding_coordinates: PaddingCoordinates, pad_v
             pad_value = tuple(pad_value)
 
         constant_values = ((pad_value, pad_value), (pad_value, pad_value), (0, 0))
+        # Fixes issue with numpy deprecation warning since constant_values is ragged array (Have to explicitly specify object dtype)
+        constant_values = np.array(constant_values, dtype=np.object_)
+
         padding_values = (pad_h, pad_w, (0, 0))
     else:
         if isinstance(pad_value, numbers.Number):