How to use

  • Install yolov5:
pip install -U yolov5
  • Load model and perform prediction:
import yolov5

# load model
model = yolov5.load('fcakyon/yolov5n-seg-v7.0')

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

# perform inference
results = model(img)
  • Finetune the model on your custom dataset:
yolov5 segment train --img 128 --weights fcakyon/yolov5n-seg-v7.0 --epochs 1 --device cpu
Downloads last month
11
Inference API
Inference API (serverless) has been turned off for this model.