"vscode:/vscode.git/clone" did not exist on "24c062aaa19f5626d03d058daf8afffa2dfd49f7"
Unverified Commit 0cc90808 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Minor perf on saturation uint8 (#6940)

parent 98f1a409
...@@ -53,7 +53,11 @@ def adjust_saturation_image_tensor(image: torch.Tensor, saturation_factor: float ...@@ -53,7 +53,11 @@ def adjust_saturation_image_tensor(image: torch.Tensor, saturation_factor: float
if c == 1: # Match PIL behaviour if c == 1: # Match PIL behaviour
return image return image
return _blend(image, _rgb_to_gray(image), saturation_factor) grayscale_image = _rgb_to_gray(image, cast=False)
if not image.is_floating_point():
grayscale_image = grayscale_image.floor_()
return _blend(image, grayscale_image, saturation_factor)
adjust_saturation_image_pil = _FP.adjust_saturation adjust_saturation_image_pil = _FP.adjust_saturation
......
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