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
No related branches found
No related tags found
1 merge request!43Tests extensions against lower case versions
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 3 deletions
bob/io/base/__init__.py
+ 3
− 3
View file @ b90e7a0d
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment