Spaces:
Sleeping
Sleeping
File size: 727 Bytes
5b889a9 24a05f6 5b889a9 60dc102 |
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 |
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-16-epoch9.keras'
num_classes = len(id_to_name)
input_shape = (1, num_frames, resolution, resolution, 3)
detect_object_frame_steps = 5
classify_action_frame_steps = 15
classify_action_num_frames = 8
yield_frame_steps = 5
|