Use two step conversion from float to uint8 to reduce platform depend…

…ence
pytorch · Aug 30, 2024 · b6d16e6 · b6d16e6
1 parent db60022
commit b6d16e6
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")