Download checkpoints (nano, small, medium, large, and x-large) from Google Drive.
pip install ultralytics
from ultralytics import YOLO
# load one of the models
model = YOLO(PATH_TO_MODEL)
# Predict with the model
results = model.predict(PATH_TO_IMAGE, save=True)