Fix functional.adjust_gamma (#5427)

8bf46d4e · Nikita Shulga · GitHub · bc0da3ed · 8bf46d4e
Unverified Commit 8bf46d4e authored Feb 16, 2022 by Nikita Shulga Committed by GitHub Feb 16, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/transforms/functional_pil.py torchvision/transforms/functional_pil.py +1 -1

No files found.
--- a/torchvision/transforms/functional_pil.py
+++ b/torchvision/transforms/functional_pil.py
@@ -119,7 +119,7 @@ def adjust_gamma(

    input_mode = img.mode
    img = img.convert("RGB")
-    gamma_map = [(255 + 1 - 1e-3) * gain * pow(ele / 255.0, gamma) for ele in range(256)] * 3
+    gamma_map = [int((255 + 1 - 1e-3) * gain * pow(ele / 255.0, gamma)) for ele in range(256)] * 3
    img = img.point(gamma_map)  # use PIL's point-function to accelerate this part

    img = img.convert(input_mode)