From b90e7a0d97b7d219aefbe61e8f1d97fe32ca27df Mon Sep 17 00:00:00 2001 From: Amir MOHAMMADI <amir.mohammadi@idiap.ch> Date: Wed, 18 May 2022 19:49:20 +0200 Subject: [PATCH] Test for lower case version of extensions --- bob/io/base/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bob/io/base/__init__.py b/bob/io/base/__init__.py index 0e7a19f..5b1d23c 100644 --- a/bob/io/base/__init__.py +++ b/bob/io/base/__init__.py @@ -18,7 +18,6 @@ ImageFile.LOAD_TRUNCATED_IMAGES = True hdf5_extensions = [".hdf5", ".h5", ".hdf", ".hdf5", ".h5", ".hdf", ".hdf5"] image_extensions = [ ".jpg", - ".JPG", ".jpeg", ".png", ".bmp", @@ -82,8 +81,8 @@ def open_file(filename): img = img[:, :, 0] return img - extension = os.path.splitext(filename)[1] # get the extension - # logger.error(filename) + # get the extension + extension = os.path.splitext(filename)[1].lower() if extension in hdf5_extensions: with h5py.File(filename, "r") as f: @@ -102,6 +101,7 @@ def open_file(filename): if string_dtype is not None: dataset = dataset.asstr() return dataset[()] + elif extension in image_extensions: from ..image import to_bob -- GitLab