import cv2
import numpy as np
from ultralytics import YOLO

model = YOLO("yolov8n-seg.pt")

results = model.track(
    "input.mp4",
    conf=0.4,
    persist=True,
    stream=True,
    tracker="app.yolo.tracker.yml",
)
for result in results:
    print("image orig shape =", result.orig_shape[0], result.orig_shape[1])
    print("image orig data  =", len(result.orig_img), len(result.orig_img[0]), type(result.orig_img))
    print("ids  =", result.boxes.id)
    print("cls  =", result.boxes.cls)
    print("conf =", result.boxes.conf)
    print("mask =", len(result.masks))

