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 ...@@ -18,7 +18,6 @@ ImageFile.LOAD_TRUNCATED_IMAGES = True
hdf5_extensions = [".hdf5", ".h5", ".hdf", ".hdf5", ".h5", ".hdf", ".hdf5"] hdf5_extensions = [".hdf5", ".h5", ".hdf", ".hdf5", ".h5", ".hdf", ".hdf5"]
image_extensions = [ image_extensions = [
".jpg", ".jpg",
".JPG",
".jpeg", ".jpeg",
".png", ".png",
".bmp", ".bmp",
...@@ -82,8 +81,8 @@ def open_file(filename): ...@@ -82,8 +81,8 @@ def open_file(filename):
img = img[:, :, 0] img = img[:, :, 0]
return img return img
extension = os.path.splitext(filename)[1] # get the extension # get the extension
# logger.error(filename) extension = os.path.splitext(filename)[1].lower()
if extension in hdf5_extensions: if extension in hdf5_extensions:
with h5py.File(filename, "r") as f: with h5py.File(filename, "r") as f:
...@@ -102,6 +101,7 @@ def open_file(filename): ...@@ -102,6 +101,7 @@ def open_file(filename):
if string_dtype is not None: if string_dtype is not None:
dataset = dataset.asstr() dataset = dataset.asstr()
return dataset[()] return dataset[()]
elif extension in image_extensions: elif extension in image_extensions:
from ..image import to_bob 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