From aa45449197e1d5c7d0510ed22e4caf07f97ea45b Mon Sep 17 00:00:00 2001
From: Kentaro Wada <www.kentaro.wada@gmail.com>
Date: Thu, 9 Jan 2020 11:44:16 +0000
Subject: [PATCH] Add labelme.utils.img_data_to_arr

---
 labelme/utils/__init__.py |  1 +
 labelme/utils/image.py    | 10 ++++++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/labelme/utils/__init__.py b/labelme/utils/__init__.py
index 31b12c2..d1dde4d 100644
--- a/labelme/utils/__init__.py
+++ b/labelme/utils/__init__.py
@@ -5,6 +5,7 @@ from ._io import lblsave
 from .image import apply_exif_orientation
 from .image import img_arr_to_b64
 from .image import img_b64_to_arr
+from .image import img_data_to_arr
 from .image import img_data_to_png_data
 
 from .shape import labelme_shapes_to_label
diff --git a/labelme/utils/image.py b/labelme/utils/image.py
index ab7e45d..7bc07cb 100644
--- a/labelme/utils/image.py
+++ b/labelme/utils/image.py
@@ -7,13 +7,19 @@ import PIL.Image
 import PIL.ImageOps
 
 
-def img_b64_to_arr(img_b64):
+def img_data_to_arr(img_data):
     f = io.BytesIO()
-    f.write(base64.b64decode(img_b64))
+    f.write(img_data)
     img_arr = np.array(PIL.Image.open(f))
     return img_arr
 
 
+def img_b64_to_arr(img_b64):
+    img_data = base64.b64decode(img_b64)
+    img_arr = img_data_to_arr(img_data)
+    return img_arr
+
+
 def img_arr_to_b64(img_arr):
     img_pil = PIL.Image.fromarray(img_arr)
     f = io.BytesIO()
-- 
GitLab