Vehicle-cpp/hkpc/STIP/classify_predict.py

from ultralytics import YOLO

# Load a model
model = YOLO('models/classify/best.pt')  # pretrained YOLOv8n model

# Run batched inference on a list of images
results = model('data/7.jpg')  # return a list of Results objects

# Process results list
for result in results:
    probs = result.probs  # Probs object for classification outputs
    print(probs)