yolo val classify data=path/to/ImageNet device=0
yolo val classify data=path/to/ImageNet batch=1 device=0|cpu
YOLO11n-clsをMNIST160データセットに画像サイズ64で100エポック学習させる。利用可能な引数の全リストは設定ページを参照。
例
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-cls.yaml") # build a new model from YAML
model = YOLO("yolo11n-cls.pt") # load a pretrained model (recommended for training)
model = YOLO("yolo11n-cls.yaml").load("yolo11n-cls.pt") # build from YAML and transfer weights
# Train the model
results = model.train(data="mnist160", epochs=100, imgsz=64)
# Build a new model from YAML and start training from scratch
yolo classify train data=mnist160 model=yolo11n-cls.yaml epochs=100 imgsz=64
# Start training from a pretrained *.pt model
yolo classify train data=mnist160 model=yolo11n-cls.pt epochs=100 imgsz=64
# Build a new model from YAML, transfer pretrained weights to it and start training
yolo classify train data=mnist160 model=yolo11n-cls.yaml pretrained=yolo11n-cls.pt epochs=100 imgsz=64
YOLO 分類データセットのフォーマットについては、「データセットガイド」を参照してください。
学習済みYOLO11n-clsモデルの検証 精度 を MNIST160 データセットに適用した。引数は必要ありません。 model
トレーニング data
と引数をモデル属性として使用する。
例
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-cls.pt") # load an official model
model = YOLO("path/to/best.pt") # load a custom model
# Validate the model
metrics = model.val() # no arguments needed, dataset and settings remembered
metrics.top1 # top1 accuracy
metrics.top5 # top5 accuracy
訓練されたYOLO11n-clsモデルを使って、画像の予測を実行する。
例
詳細を見る predict
モードの詳細は 予測する ページを参照されたい。
YOLO11n-clsモデルを、ONNX 、CoreML などの異なるフォーマットにエクスポートします。
例
利用可能なYOLO11-clsのエクスポート形式は、以下の表の通りです。どのフォーマットでも format
引数、すなわち format='onnx'
または format='engine'
.エクスポートされたモデルを直接予測または検証することができます。 yolo predict model=yolo11n-cls.onnx
.使用例は、エクスポート完了後に表示されます。
フォーマット | format 議論 | モデル | メタデータ | 議論 |
---|---|---|---|---|
PyTorch | - | yolo11n-cls.pt | ✅ | - |
TorchScript | torchscript | yolo11n-cls.torchscript | ✅ | imgsz , optimize , batch |
ONNX | onnx | yolo11n-cls.onnx | ✅ | imgsz , half , dynamic , simplify , opset , batch |
OpenVINO | openvino | yolo11n-cls_openvino_model/ | ✅ | imgsz , half , int8 , batch |
TensorRT | engine | yolo11n-cls.engine | ✅ | imgsz , half , dynamic , simplify , workspace , int8 , batch |
CoreML | coreml | yolo11n-cls.mlpackage | ✅ | imgsz , half , int8 , nms , batch |
TF SavedModel | saved_model | yolo11n-cls_saved_model/ | ✅ | imgsz , keras , int8 , batch |
TF GraphDef | pb | yolo11n-cls.pb | ❌ | imgsz , batch |
TF ライト | tflite | yolo11n-cls.tflite | ✅ | imgsz , half , int8 , batch |
TF エッジTPU | edgetpu | yolo11n-cls_edgetpu.tflite | ✅ | imgsz |
TF.js | tfjs | yolo11n-cls_web_model/ | ✅ | imgsz , half , int8 , batch |
PaddlePaddle | paddle | yolo11n-cls_paddle_model/ | ✅ | imgsz , batch |
MNN | mnn | yolo11n-cls.mnn | ✅ | imgsz , batch , int8 , half |
NCNN | ncnn | yolo11n-cls_ncnn_model/ | ✅ | imgsz , half , batch |
IMX500 | imx | yolo11n-cls_imx_model/ | ✅ | imgsz , int8 |
詳細を見る export
詳細は 輸出 ページを参照されたい。
YOLO11モデル yolo11n-cls.pt
は、効率的な画像分類のために設計されている。これは、信頼度スコアとともに画像全体に単一のクラス・ラベルを割り当てる。これは、画像内のオブジェクトの位置や形状を特定するのではなく、画像の特定のクラスを知るだけで十分なアプリケーションに特に有用である。
YOLO11モデルをトレーニングするには、Python またはCLI コマンドのいずれかを使用する。例えば yolo11n-cls
モデルをMNIST160データセットに適用し、画像サイズ64で100エポック行った:
例
その他の設定オプションについては、設定のページをご覧ください。
事前に訓練されたYOLO11分類モデルは モデル セクションをご覧ください。以下のようなモデル yolo11n-cls.pt
, yolo11s-cls.pt
, yolo11m-cls.pt
などは イメージネット データセットは簡単にダウンロードでき、様々な画像分類タスクに使用できる。
Python 、CLI コマンドを使って、学習済みYOLO11モデルを様々なフォーマットにエクスポートすることができます。例えば、ONNX フォーマットにモデルをエクスポートする:
例
詳細なエクスポートオプションについては、エクスポートのページを参照してください。
MNIST160のようなデータセットで学習済みモデルの精度を検証するには、以下のPython またはCLI コマンドを使用できます:
例
詳しくはValidateセクションをご覧ください。