13 lines
519 B
Python
13 lines
519 B
Python
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)
|