minh9972t12 commited on
Commit
a88bcc9
·
1 Parent(s): eda53ce

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +29 -1
config.yaml CHANGED
@@ -4,6 +4,18 @@ model:
4
  confidence: 0.39 # Confidence threshold
5
  iou_threshold: 0.45 # NMS IoU threshold
6
  device: "cpu" # cuda:0 for GPU, cpu for CPU
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
  # Detection Configuration
9
  detection:
@@ -28,4 +40,20 @@ visualization:
28
  colors:
29
  existing_damage: [0, 255, 0] # Green for existing damages
30
  new_damage: [0, 0, 255] # Red for new damages
31
- matched_damage: [255, 255, 0] # Yellow for matched damages
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  confidence: 0.39 # Confidence threshold
5
  iou_threshold: 0.45 # NMS IoU threshold
6
  device: "cpu" # cuda:0 for GPU, cpu for CPU
7
+
8
+ # Thêm cấu hình auto-reload
9
+ auto_reload:
10
+ enabled: true
11
+ max_inferences: 100 # Reload sau 100 lần inference
12
+ max_hours: 2 # Reload sau 2 giờ
13
+ memory_threshold_mb: 1000 # Reload nếu GPU memory > 1GB
14
+
15
+ # Backup model paths (fallback nếu model chính bị lỗi)
16
+ backup_models:
17
+ - "models/last.pt"
18
+ - "models/best.onnx"
19
 
20
  # Detection Configuration
21
  detection:
 
40
  colors:
41
  existing_damage: [0, 255, 0] # Green for existing damages
42
  new_damage: [0, 0, 255] # Red for new damages
43
+ matched_damage: [255, 255, 0] # Yellow for matched damages
44
+
45
+ # Monitoring Configuration
46
+ monitoring:
47
+ enabled: true
48
+ log_memory_usage: true
49
+ log_inference_time: true
50
+ alert_thresholds:
51
+ memory_usage_percent: 85
52
+ inference_time_ms: 5000
53
+
54
+ # Performance Optimization
55
+ performance:
56
+ enable_gpu_cache_clearing: true
57
+ batch_size: 1
58
+ num_threads: 4
59
+ enable_mixed_precision: false # Để false nếu có vấn đề stability