Model Description

Ultralytics: YOLOv8 in PyTorch > ONNX > CoreML > TFLite

Installation

pip install ultralytics

Yolov8 Inference

from ultralytics import YOLO

model = YOLO('techzizou/yolov8n')
model.conf = conf_threshold
model.iou = iou_threshold
prediction = model.predict(image, imgsz=image_size, show=False, save=False)