14 lines
No EOL
583 B
Python
14 lines
No EOL
583 B
Python
from ultralytics import YOLO
|
|
|
|
|
|
# Load an official or custom model
|
|
# model = YOLO('yolov8n.pt') # Load an official Detect model
|
|
model = YOLO('yolov8s.pt') # Load an official Detect model
|
|
# model = YOLO('yolov8n-seg.pt') # Load an official Segment model
|
|
# model = YOLO('yolov8n-pose.pt') # Load an official Pose model
|
|
# model = YOLO('path/to/best.pt') # Load a custom trained model
|
|
|
|
|
|
# Run inference on 'bus.jpg' with arguments
|
|
for i in model.predict("D:\Projects\FaceTrackingCamerav3\Raymond.mp4", save=True, save_frames=True, save_txt=True, conf=0.5, stream=True):
|
|
continue |