| from ultralytics import YOLO | |
| blankModel = YOLO('Models/blankDet_v1.1.pt') | |
| frameModel = YOLO('Models/seg_v1.1.pt') | |
| frame_shape = (416, 416) # Set the input frame shape | |
| EXPAND_RATIO = 0.1 # Inner padding | |
| OUTER_PADDING_SIZE = 5 # Outer padding size | |
| seg_conf = 0.60 | |
| det_conf = 0.50 | |
| classes_to_delete = ['ghw'] | |
| expected_segments = ['ff', 'bg', 'sw', 'fns'] | |