yolov8-segment-trash-detection

作者 turhancan97 · 下载 0 · 点赞 0 · 更新 2023-11-03

computer-vision yolov8 segmentation

Model Description

Ultralytics: YOLOv8 in PyTorch > ONNX > CoreML > TFLite]

Installation

pip install ultralytics

Yolov8 Inference

from ultralytics import YOLO

model = YOLO('turhancan97/yolov8-segment-trash-detection')
prediction = model.predict(image, imgsz=image_size, show=False, save=False)