From c74a37a5f2116bb81f87effb10ed2fd81448192c Mon Sep 17 00:00:00 2001 From: matzek92 Date: Tue, 15 Nov 2022 18:31:14 +0100 Subject: [PATCH] Update data_loader.py imageio.core.util.Array conversion to ndarray. --- keras_segmentation/data_utils/data_loader.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/keras_segmentation/data_utils/data_loader.py b/keras_segmentation/data_utils/data_loader.py index 3dc0ef7ac..cdc1e2d92 100755 --- a/keras_segmentation/data_utils/data_loader.py +++ b/keras_segmentation/data_utils/data_loader.py @@ -4,6 +4,7 @@ import six import numpy as np import cv2 +import imageio try: from collections.abc import Sequence @@ -142,6 +143,8 @@ def get_image_array(image_input, raise DataLoaderError("get_image_array: path {0} doesn't exist" .format(image_input)) img = cv2.imread(image_input, read_image_type) + elif isinstance(image_input, imageio.core.util.Array): + img = np.asarray(image_input) else: raise DataLoaderError("get_image_array: Can't process input type {0}" .format(str(type(image_input))))