Spaces:
Sleeping
Sleeping
Commit
·
a88bcc9
1
Parent(s):
eda53ce
Update config.yaml
Browse files- 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
|