From 11a70614890b2cd8ad9f676992dda80547b78a5d Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Wed, 18 May 2022 19:49:48 +0200
Subject: [PATCH] Fix bob_to_pillow float image conversion
---
bob/io/image.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/bob/io/image.py b/bob/io/image.py
index 6776e3c..682a454 100644
--- a/bob/io/image.py
+++ b/bob/io/image.py
@@ -64,8 +64,10 @@ def bob_to_pillow(img):
img = to_matplotlib(img)
# if img is floating point, convert to uint8
if isinstance(img, np.floating):
- # we expect float images to be between 0 and 1
- img = np.clip(img, 0, 1) * 255
+ # In Bob, we expect float images to be between 0 and 255
+ # see https://gitlab.idiap.ch/bob/bob.bio.face/-/issues/48
+ img = np.round(img)
+ img = np.clip(img, 0, 255)
img = img.astype(np.uint8)
return Image.fromarray(img)
@@ -108,7 +110,7 @@ def opencvbgr_to_bob(img):
If the image dimension is less than 3.
"""
if img.ndim < 3:
- raise ValueError("You need to provide at least a 3 dimensional image")
+ return img
img = img[..., ::-1]
return to_bob(img)
@@ -134,7 +136,7 @@ def bob_to_opencvbgr(img):
If the image dimension is less than 3.
"""
if img.ndim < 3:
- raise ValueError("You need to provide at least a 3 dimensional image")
+ return img
img = img[..., ::-1, :, :]
return to_matplotlib(img)
--
GitLab