Yolo 深圳
深セン
今すぐ参加

Ultralytics YOLO モデルのトレーニング、検証、予測、エクスポート、ベンチマーク方法

Nuvola Ladi

3分で読めます

2024年7月24日

Ultralytics YOLO モデルのトレーニング、検証、予測、エクスポート、ベンチマークの方法をご紹介します!

Ultralytics 世界に飛び込んで、さまざまなYOLO モデルで利用可能なさまざまなモードを探ってみましょう。カスタムオブジェクト検出モデルをトレーニングするにしても、セグメンテーションに取り組むにしても、これらのモードを理解することは非常に重要なステップです。さっそく飛び込んでみましょう!

Ultralytics ドキュメントにはトレーニング検証 予測エクスポートベンチマーク追跡など、モデルに利用できるいくつかのモードが記載されています。 track.これらのモードはそれぞれユニークな目的を持ち、モデルのパフォーマンスとデプロイメントを最適化するのに役立ちます。

学習モード

まず、trainモードを見てみましょう。ここでは、モデルを構築および改良します。詳細な手順とビデオガイドはドキュメントに記載されており、カスタムモデルのトレーニングを簡単に開始できます。

モデルのトレーニングでは、モデルに新しいデータセットを与え、様々なパターンを学習させる。一度学習されたモデルは、学習された新しいオブジェクトをdetect するためにリアルタイムで使用することができる。学習プロセスを開始する前に、YOLO 形式でデータセットに注釈を付けることが不可欠だ。

検証モード

次に、検証モードに入りましょう。検証は、ハイパーパラメータをチューニングし、モデルの性能を保証するために不可欠です。Ultralytics 、自動設定、マルチメトリックのサポート、Python APIとの互換性など、さまざまな検証オプションが用意されています。以下のコマンドで、コマンドラインインターフェイスCLI)から直接検証を実行することもできます。

なぜ検証するのか?

検証は以下にとって重要です:

  • Precision(精度):モデルがオブジェクトを正確に検出することを保証します。
  • 利便性: 検証プロセスを効率化します。
  • 柔軟性: 複数の検証方法を提供します。
  • ハイパーパラメータ調整: より良いパフォーマンスのためにモデルを最適化します。

Ultralytics 、Python スクリプトにコピー&ペーストできるユーザー例も提供しています。これらの例には、画像サイズ、バッチサイズ、デバイスCPU またはGPU)、交差オーバーユニオンIoU)などのパラメータが含まれています。

予測モード

モデルの学習と検証が完了したら、いよいよ予測を行います。Predictモードでは、新しいデータに対して推論を実行し、モデルの動作を確認することができます。このモードは、実際のデータでモデルのパフォーマンスをテストするのに最適です。

以下のpython コード・スニペットで、画像に対して予測を実行できるようになります!

エクスポートモード

検証と予測の後、モデルをデプロイしたくなるかもしれません。エクスポート・モードにより、モデルをONNX TensorRT様々なフォーマットに変換することができ、異なるプラットフォームへのデプロイが容易になります。

ベンチマークモード

最後に、ベンチマークモードがあります。ベンチマークは、さまざまなシナリオでモデルのパフォーマンスを評価するために不可欠です。このモードは、リソースの割り当て、最適化、およびコスト効率に関する情報に基づいた意思決定を行うのに役立ちます。

ベンチマークの方法

ベンチマークを実行するには、ドキュメントで提供されているユーザー例を使用できます。これらの例では、ONNX TensorRT主要なメトリクスやエクスポート形式をカバーしています。また、整数量子化(INT8)や浮動小数点量子化(FP16)などのパラメータを指定して、設定の違いによる性能への影響を確認することもできます。

実世界のベンチマーク例

ベンチマークの実例を見てみましょう。PyTorch モデルをベンチマークしたところ、RTX 3070GPU推論速度が68ミリ秒だった。TorchScriptエクスポートすると、推論速度は4ミリ秒に低下し、大幅な改善が見られます。

ONNX モデルの推論速度は21ミリ秒である。これらのモデルをCPU (Intel i9 第13世代)でテストしたところ、さまざまな結果が得られた。TorchScript 115ミリ秒で動作し、ONNX 84ミリ秒と良好な結果でした。最後に、Intel ハードウェアに最適化されたOpenVINO 、23ミリ秒という驚異的な速度を達成した。

__wf_reserved_inherit
図1.Ultralytics YOLO モデルによるベンチマークの実行方法を実演するニコライ・ニールセン

ベンチマークの重要性

ベンチマークは、異なるハードウェアおよびエクスポート形式がモデルのパフォーマンスにどのように影響するかを示します。特にカスタムハードウェアまたはエッジデバイスにデプロイする場合は、モデルをベンチマークすることが重要です。このプロセスにより、モデルがターゲット環境に最適化され、可能な限り最高のパフォーマンスが提供されます。

結論

要約すると、Ultralytics ドキュメントのモードは、YOLO モデルのトレーニング、検証、予測、エクスポート、およびベンチマークを行うための強力なツールです。各モードは、モデルを最適化し、展開の準備をする上で重要な役割を果たします。

ぜひ、当社のコミュニティを探求し、参加して、提供されているコードスニペットをあなたのプロジェクトで試してみてください。これらのツールを使用することで、高性能なモデルを作成し、あらゆる環境で効率的に実行できるようになります。

AIの未来を
共に築きましょう!

未来の機械学習で、新たな一歩を踏み出しましょう。

無料ではじめる