TrafficSignDetector / config.yaml
VietCat's picture
Add confidence threshold slider and update class labels
81a35ad
model:
path: 'VietCat/GTSRB-Model/models/GTSRB.pt' # Path to the YOLO model on Hugging Face Hub (will be downloaded automatically)
confidence_threshold: 0.30 # Minimum confidence for detections (0.3 filters most false positives)
inference:
box_color: (128, 0, 128) # Purple color for bounding boxes (BGR format)
text_color: (255, 255, 255) # White color for labels
thickness: 2 # Thickness of bounding box lines
classes:
- 'one_way_prohibition'
- 'no_parking'
- 'no_stopping_and_parking'
- 'no_turn_left'
- 'no_turn_right'
- 'no_u_turn'
- 'no_u_and_left_turn'
- 'no_u_and_right_turn'
- 'no_motorbike_entry_turning'
- 'no_car_entry_turning'
- 'no_truck_entry_turning'
- 'other_prohibition'
- 'indication'
- 'direction'
- 'speed_limit'
- 'weight_limit'
- 'height_limit'
- 'pedestrian_crossing'
- 'intersection_danger'
- 'road_danger'
- 'pedestrian_danger'
- 'construction_danger'
- 'slow_warning'
- 'other_warning'
- 'vehicle_permission_lane'
- 'vehicle_and_speed_permission_lane'
- 'overpass_route'
- 'no_more_prohibition'
- 'other'