Spaces:
Sleeping
Sleeping
# 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 | |