# Data
data_dir = 'storage/dataset'
training_ratio = 0.7
validation_ratio = 0.02
num_frames = 8
frame_step = 1
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
model_id = 'a0'
checkpoint_dir = f'storage/pretrained_weights/movinet_{model_id}_base'
num_classes = 6

# Inference
detect_object_frame_steps = 5
classify_action_frame_steps = 15
classify_action_num_frames = 8

# Train
train_id = 8
batch_size = 16
learning_rate = 0.001
epochs = 15
model_save_path = f'storage/output/classifier-{train_id}.keras'
log_dir = f'storage/logs/classifier-{train_id}.log'

# Train more
initial_epoch = 0