Spaces:
Sleeping
Sleeping
File size: 863 Bytes
160ded7 |
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 |
# 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
|