Yolov5n Cls V7.0

🚀 YOLOv5画像分類モデル

このプロジェクトはYOLOv5を用いた画像分類モデルで、コンピュータビジョンタスクにおける画像分類に利用できます。

🚀 クイックスタート

インストール

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

📄 ライセンス

このプロジェクトはGPL-3.0ライセンスの下で提供されています。