Support webp

Author: yushulx

examples/official/annotation_tool/main.py

@@ -451,9 +451,9 @@ def dropEvent(self, event):
                 if os.path.isdir(local):
                     for root_dir, _, files in os.walk(local):
                         for f in files:
-                            if f.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif")):
+                            if f.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif", ".webp")):
                                 paths.append(os.path.join(root_dir, f))
-                elif local.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif")):
+                elif local.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif", ".webp")):
                     paths.append(local)
             if paths:
                 self.files_dropped.emit(paths)
@@ -833,7 +833,7 @@ def dropEvent(self, event):
     def _load_files(self):
         paths, _ = QFileDialog.getOpenFileNames(
             self, "Select Images", "",
-            "Images (*.png *.jpg *.jpeg *.bmp *.tiff *.tif)"
+            "Images (*.png *.jpg *.jpeg *.bmp *.tiff *.tif *.webp)"
         )
         self._add_paths(paths)
 
@@ -844,7 +844,7 @@ def _load_folder(self):
         paths = []
         for root, _, files in os.walk(folder):
             for f in files:
-                if f.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif")):
+                if f.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif", ".webp")):
                     paths.append(os.path.join(root, f))
         self._add_paths(paths)
 

examples/official/annotation_tool/viewer.py

@@ -206,7 +206,7 @@ def _on_load_files(self):
         paths = filedialog.askopenfilenames(
             title="Load Images",
             filetypes=[
-                ("Images", "*.png *.jpg *.jpeg *.bmp *.tiff *.tif"),
+                ("Images", "*.png *.jpg *.jpeg *.bmp *.tiff *.tif *.webp"),
                 ("All files", "*.*"),
             ]
         )
@@ -220,7 +220,7 @@ def _on_load_folder(self):
         paths = []
         for root_dir, _, files in os.walk(folder):
             for f in files:
-                if f.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif")):
+                if f.lower().endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif", ".webp")):
                     paths.append(os.path.join(root_dir, f))
         if paths:
             self._add_images(paths)
@@ -557,13 +557,13 @@ def _on_drop(self, event):
                 for root_dir, _, files in os.walk(p):
                     for f in files:
                         if f.lower().endswith(
-                            (".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif")
+                            (".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif", ".webp")
                         ):
                             image_files.append(os.path.join(root_dir, f))
             elif p.lower().endswith(".json"):
                 json_files.append(p)
             elif p.lower().endswith(
-                (".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif")
+                (".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".tif", ".webp")
             ):
                 image_files.append(p)
         if json_files: