minh9972t12 commited on
Commit
804a6ce
·
1 Parent(s): 8fbe7c8

Update config_version2.yaml

Browse files
Files changed (1) hide show
  1. config_version2.yaml +8 -6
config_version2.yaml CHANGED
@@ -1,7 +1,7 @@
1
  model:
2
  path: models_medium_version_2/best.pt
3
  device: cpu
4
- confidence: 0.4
5
  iou_threshold: 0.4
6
 
7
  detection:
@@ -14,11 +14,13 @@ detection:
14
  - SPECIAL_DAMAGE # index 5
15
 
16
  comparison:
17
- iou_match_threshold: 0.4
18
- position_tolerance: 50
19
- alpha: 0.6
20
- beta: 0.3
21
- gamma: 0.1
 
 
22
 
23
  visualization:
24
  line_thickness: 2
 
1
  model:
2
  path: models_medium_version_2/best.pt
3
  device: cpu
4
+ confidence: 0.25
5
  iou_threshold: 0.4
6
 
7
  detection:
 
14
  - SPECIAL_DAMAGE # index 5
15
 
16
  comparison:
17
+ iou_match_threshold: 0.35
18
+ position_tolerance: 100
19
+ alpha: 0.35
20
+ beta: 0.45
21
+ gamma: 0.2
22
+ combined_score_threshold: 0.3
23
+ reid_similarity_threshold: 0.8
24
 
25
  visualization:
26
  line_thickness: 2