Skip to content
Snippets Groups Projects
Commit b90e7a0d authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Test for lower case version of extensions

parent 7ffd44e0
Branches
Tags
1 merge request!43Tests extensions against lower case versions
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment