Add webp to list of supported files in reader.

rom1504 · Aug 3, 2023 · 148d9b0 · 148d9b0
1 parent 5408b22
commit 148d9b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/clip_retrieval/clip_inference/reader.py b/clip_retrieval/clip_inference/reader.py
@@ -22,6 +22,7 @@ def folder_to_keys(folder, enable_text=True, enable_image=True, enable_metadata=
             *path.glob("**/*.jpg"),
             *path.glob("**/*.jpeg"),
             *path.glob("**/*.bmp"),
+            *path.glob("**/*.webp"),
         ]
         image_files = {image_file.stem: image_file for image_file in image_files}
     if enable_metadata: