Fix adjust_hue on ARM

Co-authored-by: Nicolas Hug <[email protected]>
pytorch · Oct 14, 2024 · a7f0d25 · a7f0d25
1 parent ed55b03
commit a7f0d25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/transforms/_functional_pil.py b/torchvision/transforms/_functional_pil.py
@@ -110,7 +110,7 @@ def adjust_hue(img: Image.Image, hue_factor: float) -> Image.Image:
 
     np_h = np.array(h, dtype=np.uint8)
     # This will over/underflow, as desired
-    np_h += np.array(hue_factor * 255).astype(np.uint8)
+    np_h += np.int32(hue_factor * 255).astype(np.uint8)
 
     h = Image.fromarray(np_h, "L")