turhancan97's picture
Update README.md
89f1b19
|
raw
history blame
535 Bytes
metadata
license: mit
language:
  - en
datasets:
  - garythung/trashnet
  - Zesky665/TACO
  - detection-datasets/coco
tags:
  - 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)