Skip to content

GitLab

Unverified Commit 8bf46d4e authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Fix functional.adjust_gamma (#5427)

parent bc0da3ed
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/transforms/functional_pil.py
View file @ 8bf46d4e
...@@ -119,7 +119,7 @@ def adjust_gamma( ...@@ -119,7 +119,7 @@ def adjust_gamma(
input_mode = img.mode input_mode = img.mode
img = img.convert("RGB") 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.point(gamma_map) # use PIL's point-function to accelerate this part
img = img.convert(input_mode) img = img.convert(input_mode)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment