class Config:
  num_frames = 8
  frame_step = 15
  resolution = 224
  frame_size = (resolution, resolution)
  id_to_name = {
    0: 'Flying',
    1: 'Landing',
    2: 'Other',
    3: 'Straight Taxiing',
    4: 'Takeoff',
    5: 'Turning Maneuver',
  }
  name_to_id = {
    'Flying': 0,
    'Landing': 1,
    'Other': 2,
    'Straight Taxiing': 3,
    'Takeoff': 4,
    'Turning Maneuver': 5,
  }

  model_id = 'a0'
  detector_path = 'weights/yolov8n.pt'
  classifier_path = 'weights/classifier-8-epoch10.keras'
  num_classes = len(id_to_name)
  input_shape = (1, num_frames, resolution, resolution, 3)

  detect_object_frame_step = 5
  classify_action_frame_step = frame_step
  classify_action_num_frames = num_frames
  yield_frame_steps = 1