OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-31 02:32:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

torch_bgr_to_pil_image: round, don't truncate

Browse Source 
This matches what `realesrgan` does.
This commit is contained in:
Aarni Koskela 2024-01-02 17:14:05 +02:00
parent 7c3ab416ad
commit 7ad6899bf9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/upscaler_utils.py
View File

@ -30,7 +30,7 @@ def torch_bgr_to_pil_image(tensor: torch.Tensor) -> Image.Image:
# TODO: is `tensor.float().cpu()...numpy()` the most efficient idiom? # TODO: is `tensor.float().cpu()...numpy()` the most efficient idiom?
arr = tensor.float().cpu().clamp_(0, 1).numpy() # clamp arr = tensor.float().cpu().clamp_(0, 1).numpy() # clamp
arr = 255.0 * np.moveaxis(arr, 0, 2) # CHW to HWC, rescale arr = 255.0 * np.moveaxis(arr, 0, 2) # CHW to HWC, rescale
arr = arr.astype(np.uint8) arr = arr.round().astype(np.uint8)
arr = arr[:, :, ::-1] # flip BGR to RGB arr = arr[:, :, ::-1] # flip BGR to RGB
return Image.fromarray(arr, "RGB") return Image.fromarray(arr, "RGB")