Merge branch 'np2-hue' into np2

pytorch · Jun 3, 2024 · 46ef777 · 46ef777
2 parents 6012d8a + fb4fb59
commit 46ef777
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/torchvision/transforms/_functional_pil.py b/torchvision/transforms/_functional_pil.py
@@ -109,12 +109,8 @@ def adjust_hue(img: Image.Image, hue_factor: float) -> Image.Image:
     h, s, v = img.convert("HSV").split()
 
     np_h = np.array(h, dtype=np.uint8)
-    with np.errstate(over="ignore"):
-        shift = int(hue_factor * 255)
-        if shift > 0:
-            np_h += shift
-        else:
-            np_h -= abs(shift)
+    # This will over/underflow, as desired
+    np_h += np.array(hue_factor * 255).astype(np.uint8)
 
     h = Image.fromarray(np_h, "L")