from ultralytics import YOLO

# model1 = YOLO('yolov8n.pt')  # load an official model
model2 = YOLO(r'D:\1BNS_projects\yolo_safehat_num_v8\weights\best.pt')  # load a custom model

model2.predict(
    r'D:\1BNS_projects\yolo_safehat_num_v8\test_data\person.mp4',
    save=True)
# model2.predict(
#     'https://www.baidu.com/link?url=DGSn7iUIIz7OKSCQDIVGYUUJG4Z8S25fh1jSfcOE7WAltzGZt42lIAHImbi0PWMp0nrNSs4_pRLT55JZyL1CQq&wd=&eqid=873f09050212561900000005660cc4e6',
#     save=True,
#     classes=[0, 2], line_width=30)