Football_Simple_AI/config.yaml.example at main · farshidrayhancv/Football_Simple_AI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
api_keys:
  huggingface_token: ""
  roboflow_api_key: ""
ball_tracking:
  avg_frames: 5
  enable: true
  trail_length: 10
caching:
  cache_directory: ''
  enable_team_classifier_cache: true
detection:
  confidence_threshold: 0.3
  keypoint_confidence_threshold: 0.5
  nms_threshold: 0.5
  padding_ratio: 0.1
  pose_confidence_threshold: 0.5
display:
  backend: opencv
  ball_color: '#FFD700'
  original_frame_height: 600
  original_frame_width: 800
  pitch_view_height: 800
  pitch_view_width: 600
  referee_color: '#FFD700'
  segmentation_alpha: 0.6
  show_ball: true
  show_pose: true
  show_possession_detection: true
  show_segmentation: true
  show_tracking_ids: true
  show_voronoi: true
  team_colors:
    team_1: '#00BFFF'
    team_2: '#FF1493'
models:
  field_detection_model_id: football-field-detection-f07vi/14
  player_detection_model_id: football-players-detection-3zvbc/11
  pose_model: yolo11s-pose.pt
  sam_model: sam2.1_s.pt
  siglip_model_path: google/siglip-base-patch16-256
performance:
  batch_size: 32
  device: cuda
  use_gpu: true
possession_detection:
  coordinate_system: frame
  enable: true
  frame_proximity_threshold: 30
  no_possession_frames: 15
  possession_duration: 3
  possession_frames: 1
  proximity_threshold: 250
processing:
  resolution:
  - 1080
  - 1080
sahi:
  enable: false
  overlap_ratio: 0.2
  postprocess_match_threshold: 0.5
  postprocess_type: NMU
  slice_cols: 2
  slice_height: 640
  slice_rows: 2
  slice_width: 640
tracking:
  max_distance_threshold: 500
  tracker_type: bytetrack
video:
  display_fps: 30
  input_path: videos/0bfacc_0.mp4
  output_path: /home/farshid/proj/Football/videos/
  stride: 30