yolocar / config_version2.yaml
minh9972t12's picture
Update config_version2.yaml
804a6ce
raw
history blame contribute delete
658 Bytes
model:
path: models_medium_version_2/best.pt
device: cpu
confidence: 0.25
iou_threshold: 0.4
detection:
classes:
- FRONT_DAMAGE # index 0
- SIDE_DAMAGE # index 1
- REAR_DAMAGE # index 2
- SURFACE_DAMAGE # index 3
- STRUCTURAL_DAMAGE # index 4
- SPECIAL_DAMAGE # index 5
comparison:
iou_match_threshold: 0.35
position_tolerance: 100
alpha: 0.35
beta: 0.45
gamma: 0.2
combined_score_threshold: 0.3
reid_similarity_threshold: 0.8
visualization:
line_thickness: 2
font_scale: 0.5
colors:
new_damage: [0, 0, 255]
existing_damage: [0, 255, 0]
matched_damage: [255, 255, 0]