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']