Yolov5n Cls V7.0

🚀 YOLOv5圖像分類模型

本項目基於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

📄 許可證

本項目採用GPL-3.0許可證。