jc/jc-video-recognize
11
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

补充：火灾检测模型由基于YOLOv10的火灾烟雾检测模型改为复合模型[基于YOLOv8的火灾检测模型（单火焰检测）＋YOLOv10-M，专用火灾烟雾模型]

Browse Source
This commit is contained in:
zhanjiesheng
2026-06-10 14:51:14 +08:00
parent 0e011dacfd
commit ca4f977ff0
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
apps/server/services/model_service.py
View File

@@ -40,12 +40,21 @@ class ModelService:
self.model_configs = { self.model_configs = {
'fire_detection': { 'fire_detection': {
'path': os.path.join(base_dir, 'models', 'fire_detection', 'best.pt'), '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', 'type': 'yolov10',
'classes': ['Fire', 'Smoke'], 'classes': ['Fire', 'Smoke'],
'labels': {'Fire': '火焰', 'Smoke': '烟雾'}, 'labels': {'Fire': '火焰', 'Smoke': '烟雾'},
'size': '61MB', 'size': '33MB',
'description': '基于YOLOv10的火灾烟雾检测模型', 'description': '基于YOLOv10-M的火灾烟雾检测模型（来自GitHub开源项目，123K+图片训练）',
'name': '火灾检测' 'name': '烟雾检测 (YOLOv10-M)'
}, },
'helmet_detection': { 'helmet_detection': {
'path': os.path.join(base_dir, 'models', 'helmet_detection', 'yolov8n.pt'), 'path': os.path.join(base_dir, 'models', 'helmet_detection', 'yolov8n.pt'),