From ca4f977ff01da31f11207e5f80e89887e75543cf Mon Sep 17 00:00:00 2001
From: zhanjiesheng <210466500@qq.com>
Date: Wed, 10 Jun 2026 14:51:14 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85=EF=BC=9A=E7=81=AB=E7=81=BE?=
 =?UTF-8?q?=E6=A3=80=E6=B5=8B=E6=A8=A1=E5=9E=8B=E7=94=B1=E5=9F=BA=E4=BA=8E?=
 =?UTF-8?q?YOLOv10=E7=9A=84=E7=81=AB=E7=81=BE=E7=83=9F=E9=9B=BE=E6=A3=80?=
 =?UTF-8?q?=E6=B5=8B=E6=A8=A1=E5=9E=8B=E6=94=B9=E4=B8=BA=E5=A4=8D=E5=90=88?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B[=E5=9F=BA=E4=BA=8EYOLOv8=E7=9A=84=E7=81=AB?=
 =?UTF-8?q?=E7=81=BE=E6=A3=80=E6=B5=8B=E6=A8=A1=E5=9E=8B=EF=BC=88=E5=8D=95?=
 =?UTF-8?q?=E7=81=AB=E7=84=B0=E6=A3=80=E6=B5=8B=EF=BC=89=EF=BC=8BYOLOv10-M?=
 =?UTF-8?q?=EF=BC=8C=E4=B8=93=E7=94=A8=E7=81=AB=E7=81=BE=E7=83=9F=E9=9B=BE?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B]?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/server/services/model_service.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/apps/server/services/model_service.py b/apps/server/services/model_service.py
index 87c00f5..19f0e3a 100644
--- a/apps/server/services/model_service.py
+++ b/apps/server/services/model_service.py
@@ -40,12 +40,21 @@ class ModelService:
         self.model_configs = {
             'fire_detection': {
                 'path': os.path.join(base_dir, 'models', 'fire_detection', 'best.pt'),
+                'type': 'yolov8',
+                'classes': ['Fire'],
+                'labels': {'Fire': '火焰'},
+                'size': '22MB',
+                'description': '基于YOLOv8的火焰检测模型',
+                'name': '火焰检测'
+            },
+            'smoke_detection': {
+                'path': os.path.join(base_dir, 'models', 'fire_detection', 'yolov10_fire_smoke_best.pt'),
                 'type': 'yolov10',
                 'classes': ['Fire', 'Smoke'],
                 'labels': {'Fire': '火焰', 'Smoke': '烟雾'},
-                'size': '61MB',
-                'description': '基于YOLOv10的火灾烟雾检测模型',
-                'name': '火灾检测'
+                'size': '33MB',
+                'description': '基于YOLOv10-M的火灾烟雾检测模型（来自GitHub开源项目，123K+图片训练）',
+                'name': '烟雾检测 (YOLOv10-M)'
             },
             'helmet_detection': {
                 'path': os.path.join(base_dir, 'models', 'helmet_detection', 'yolov8n.pt'),