许可证: gpl-3.0
推理: 不支持
标签:

• 图像分类
• 计算机视觉
• 视觉
• yolo
• yolov5

使用方法

• 安装yolov5:

pip install -U yolov5  

• 加载模型并进行预测:

import yolov5  

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

# 设置图像  
img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'  

# 执行推理  
results = model(img)  

• 在自定义数据集上微调模型:

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