From d45790e00faa33421f71ca65a29c3c1cc5a032e7 Mon Sep 17 00:00:00 2001
From: leccelecce <24962424+leccelecce@users.noreply.github.com>
Date: Wed, 19 Mar 2025 12:30:23 +0000
Subject: [PATCH] Allow png/jpg/jpeg as well as webp

---
 frigate/api/classification.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/frigate/api/classification.py b/frigate/api/classification.py
index 632214930..b9d714ca3 100644
--- a/frigate/api/classification.py
+++ b/frigate/api/classification.py
@@ -39,7 +39,10 @@ def get_faces():
         face_dict[name] = []
 
         for file in sorted(
-            filter(lambda f: (f.endswith(".webp")), os.listdir(face_dir)),
+            filter(
+                lambda f: (f.lower().endswith((".webp", ".png", ".jpg", ".jpeg"))),
+                os.listdir(face_dir),
+            ),
             key=lambda f: os.path.getctime(os.path.join(face_dir, f)),
             reverse=True,
         ):