如何使用

• 安装yolov5:

pip install -U yolov5

• 加载模型并进行预测:

import yolov5

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

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

# perform inference
results = model(img)

• 在您的自定义数据集上进行微调:

yolov5 classify train --img 128 --data mnist2560 --model fcakyon/yolov5n-cls-v7.0 --epochs 1 --device cpu