From c620cadee3fa8729a6a4fb9bd7b8e076be9a2a2c Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Tue, 15 Feb 2022 21:34:02 -0800 Subject: [PATCH] Fix functional.adjust_gamma --- torchvision/transforms/functional_pil.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/functional_pil.py b/torchvision/transforms/functional_pil.py index fb70d1969dc..0a48a381cc7 100644 --- 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)