ultralyticsplus 0.0.4

HuggingFace utilities for Ultralytics/YOLOv8.

ultralytics+

Extra features for ultralytics/ultralytics.

installation

pip install ultralyticsplus

push to 🤗 hub

ultralyticsplus --exp_dir runs/detect/train --hf_model_id HF_USERNAME/MODELNAME

load from 🤗 hub

from ultralyticsplus import YOLO, render_predictions

# load model
model = YOLO('HF_USERNAME/MODELNAME')

# set model parameters
model.overrides['conf'] = 0.25  # NMS confidence threshold
model.overrides['iou'] = 0.45  # NMS IoU threshold
model.overrides['agnostic_nms'] = False  # NMS class-agnostic
model.overrides['max_det'] = 1000  # maximum number of detections per image

# set image
img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'

# perform inference
for result in model.predict(img, imgsz=640, return_outputs=True):
    print(result["det"]) # [[x1, y1, x2, y2, conf, class]]
    render = render_predictions(model, img=img, det=result["det"])
    render.show()