torch_bgr_to_pil_image: round, don't truncate

This matches what `realesrgan` does.
AUTOMATIC1111 · Jan 2, 2024 · 7ad6899 · 7ad6899
1 parent 7c3ab41
commit 7ad6899
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/upscaler_utils.py b/modules/upscaler_utils.py
@@ -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?
     arr = tensor.float().cpu().clamp_(0, 1).numpy()  # clamp
     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
     return Image.fromarray(arr, "RGB")