Skip to content

Commit af52d09

Browse files
authored
🐞 fix(data): Fixed comparison of file extensions for Folder dataset (#3059)
* Fixed Folder dataset to use upper case file extension * Convert provided extensions to lower case
1 parent 9ee10fd commit af52d09

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/anomalib/data/utils/path.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,17 @@ def _prepare_files_labels(
103103
if extensions is None:
104104
extensions = IMG_EXTENSIONS
105105

106+
# convert extensions to lowercase for case-insensitive matching
107+
extensions = tuple(ext.lower() for ext in extensions)
108+
106109
if isinstance(extensions, str):
107110
extensions = (extensions,)
108111

109112
if not all(extension.startswith(".") for extension in extensions):
110113
msg = f"All extensions {extensions} must start with the dot"
111114
raise RuntimeError(msg)
112115

113-
filenames = [f for f in path.glob("**/*") if f.suffix in extensions and not f.is_dir()]
116+
filenames = [f for f in path.glob("**/*") if f.suffix.lower() in extensions and not f.is_dir()]
114117
# list of files that are in hidden directories or are hidden files themselves
115118
hidden_files = [f for f in filenames if any(part.startswith(".") for part in f.parts)]
116119
if hidden_files:

0 commit comments

Comments
 (0)