File size: 358 Bytes
cb90fd0 | 1 2 3 4 5 6 7 8 9 10 11 12 | 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']
|