From fb4fb59d6acb154361d93828aa36c68eacb7d0c9 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 3 Jun 2024 11:33:00 +0100 Subject: [PATCH] Use better way from warning --- torchvision/transforms/_functional_pil.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/torchvision/transforms/_functional_pil.py b/torchvision/transforms/_functional_pil.py index 46f928f541a..527879bb6f1 100644 --- 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")