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'),