aero-recognize / configuration.py
chiyoi's picture
Tidy
ba5f5cf
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 = 8
classify_action_frame_step = frame_step
classify_action_num_frames = num_frames
yield_frame_steps = 4