YOLOv8 実験の追跡と可視化の強化Weights & Biases

のような物体検出モデルは Ultralytics YOLOv8は、多くのコンピュータ・ビジョン・アプリケーションに不可欠なものとなっている。しかし、このような複雑なモデルのトレーニング、評価、導入にはいくつかの課題があります。主要なトレーニングメトリクスの追跡、モデルバリアントの比較、モデル動作の分析、問題の検出には、大規模な計測と実験管理が必要です。

このガイドでは、次の内容を紹介します。 Ultralytics YOLOv8 との統合 Weights & Biases' を使用して、実験の追跡、モデルのチェックポイント設定、およびモデルのパフォーマンスの視覚化を強化します。また、統合の設定、トレーニング、微調整、および結果の視覚化の手順も含まれています。 Weights & Biases' インタラクティブ機能。

Weights & Biases

Weights & Biasesは、機械学習実験の追跡、可視化、管理のために設計された最先端のMLOpsプラットフォームです。実験の完全な再現性を実現するトレーニングメトリクスの自動ロギング、合理化されたデータ分析のためのインタラクティブなUI、様々な環境にデプロイするための効率的なモデル管理ツールを備えています。

YOLOv8 トレーニングWeights & Biases

YOLOv8 トレーニング・プロセスの効率化と自動化のために、Weights & Biases 。

インストール

必要なパッケージをインストールするには

# Install the required packages for YOLOv8 and Weights & Biases
pip install --upgrade ultralytics==8.0.186 wandb

インストールプロセスに関する詳細な説明とベストプラクティスについては、YOLOv8 インストールガイドをご覧ください。YOLOv8 に必要なパッケージをインストールする際、何らかの問題が発生した場合は、解決策やヒントについて「よくある問題」ガイドを参照してください。

設定Weights & Biases

必要なパッケージをインストールしたら、次のステップはWeights & Biases 環境のセットアップです。これにはWeights & Biases アカウントの作成と、開発環境と W&B プラットフォームをスムーズに接続するために必要な API キーの取得が含まれます。

まず、ワークスペースのWeights & Biases 環境を初期化します。以下のコマンドを実行し、プロンプトの指示に従ってください。

# Initialize your Weights & Biases environment
import wandb
wandb.login()

Weights & Biases 認証ページに移動し、API キーを作成して取得します。このキーを使用して、W&B でお客様の環境を認証します。

使い方トレーニングYOLOv8 Weights & Biases

Weights & Biases を使ったYOLOv8 モデル・トレーニングの使い方の説明に入る前に、 Ultralytics が提供するYOLOv8 モデルの数々を必ずチェックしてください。これは、あなたのプロジェクトの要件に最も適切なモデルを選択するのに役立ちます。

import wandb
from wandb.integration.ultralytics import add_wandb_callback

from ultralytics import YOLO

# Initialize a Weights & Biases run
wandb.init(project="ultralytics", job_type="training")

# Load a YOLO model
model = YOLO("yolov8n.pt")

# Add W&B Callback for Ultralytics
add_wandb_callback(model, enable_model_checkpointing=True)

# Train and Fine-Tune the Model
model.train(project="ultralytics", data="coco8.yaml", epochs=5, imgsz=640)

# Validate the Model
model.val()

# Perform Inference and Log Results
model(["path/to/image1", "path/to/image2"])

# Finalize the W&B Run
wandb.finish()

コードを理解する

上記の使用コードスニペットで紹介されている手順を理解しましょう。

• ステップ 1：Weights & Biases ランの初期化：プロジェクト名とジョブタイプを指定し、Weights & Biases 実行を初期化することから始める。このランは、モデルのトレーニングおよび検証プロセスを追跡および管理します。

• ステップ 2:YOLOv8 モデルとデータセットを定義する：使用したいモデル・バリアントとデータセットを指定する。YOLO モデルが、指定されたモデル・ファイルで初期化されます。

• ステップ 3：Weights & Biases Ultralytics のコールバックを追加する：このステップは、トレーニング・メトリクスと検証結果の自動ロギングをWeights & Biases 、モデルのパフォーマンスの詳細なビューを提供するため、非常に重要です。

• ステップ4：モデルの訓練と微調整：指定されたデータセット、エポック数、画像サイズでモデルのトレーニングを開始します。トレーニング・プロセスには、各エポックの終了時にメトリクスと予測値のロギングが含まれ、モデルの学習進捗の包括的なビューを提供します。

• ステップ5：モデルの検証：トレーニングの後、モデルを検証する。このステップは、未知のデータに対するモデルのパフォーマンスを評価し、その汎化可能性を確保するために極めて重要である。

• ステップ6：推論の実行と結果のログ：モデルは指定された画像に対して予測を実行します。これらの予測は、視覚的なオーバーレイと洞察とともに、インタラクティブな探索のためにW&Bテーブルに自動的に記録されます。

• ステップ 7: W&B の実行を確定します：このステップでデータロギングを終了し、モデルのトレーニングと検証プロセスの最終状態をW&Bダッシュボードに保存します。

アウトプットを理解する

上記のコード・スニペットを実行すると、次のような主な出力が期待できる：

• トレーニングプロセスの開始を示す一意のIDを持つ新しいランのセットアップ。
• レイヤーやパラメータの数など、モデルの構造の簡潔な概要。
• 各トレーニングエポックにおいて、box loss、cls loss、dfl loss、precision、recall、mAPスコアなどの重要なメトリクスが定期的に更新される。
• トレーニングの最後には、モデルの推論速度や全体的な精度のメトリクスを含む詳細なメトリクスが表示されます。
• トレーニングプロセスの詳細な分析と視覚化のためのWeights & Biases ダッシュボードへのリンク、およびローカルのログファイルの場所に関する情報。

Weights & Biases ダッシュボードの表示

使用法のコード・スニペットを実行した後、出力にある提供されたリンクからWeights & Biases (W&B) ダッシュボードにアクセスすることができます。このダッシュボードでは、YOLOv8 を使ったモデルのトレーニングプロセスを包括的に見ることができます。

Weights & Biases ダッシュボードの主な機能

• リアルタイムのメトリクス追跡：損失、精度、検証スコアなどのメトリクスがトレーニング中にどのように変化するかを観察し、モデルチューニングのための洞察を即座に提供します。 Weights & Biases を 使用した実験の追跡方法をご覧ください。

• ハイパーパラメータの最適化:Weights & Biases は、学習率やバッチサイズなどの重要なパラメータを微調整し、YOLOv8 のパフォーマンスを向上させるのに役立ちます。

• 比較分析：このプラットフォームでは、さまざまなモデル構成の影響を評価するために不可欠な、異なるトレーニング実行の横並び比較が可能です。

• トレーニング進捗の視覚化：主要なメトリクスをグラフィカルに表示することで、エポックにわたるモデルのパフォーマンスを直感的に理解することができます。 Weights & Biases が検証結果の視覚化にどのように役立つかをご覧ください。

• リソースのモニタリング：トレーニングプロセスの効率を最適化するために、CPU 、GPU 、メモリ使用量を追跡します。

• モデル成果物の管理：モデルのチェックポイントにアクセスして共有し、展開やコラボレーションを容易にします。

• 画像オーバーレイによる推論結果の表示：Weights & Biases のインタラクティブなオーバーレイを使用して、画像上で予測結果を視覚化。Weights & Biases' 画像オーバーレイ機能の詳細については、こちらのリンクをご覧ください。 Weights & Biases' 画像オーバーレイがモデルの推論をどのように視覚化するかをご覧ください。

これらの機能を使用することで、YOLOv8 モデルのトレーニングを効果的に追跡、分析、最適化し、最高のパフォーマンスと効率を実現することができます。

概要

このガイドは、探索に役立ちました Ultralytics' YOLOv8 との統合 Weights & Biases.これは、モデルのトレーニングと予測の結果を効率的に追跡および視覚化するこの統合の機能を示しています。

使い方の詳細については、Weights & Biases'公式ドキュメントをご覧ください。

また、Ultralytics の統合ガイドページで、さまざまなエキサイティングな統合について詳しくご覧ください。

よくあるご質問

YOLOv8 とWeights & Biases に必要なパッケージのインストール方法は？

YOLOv8 とWeights & Biases に必要なパッケージをインストールするには、コマンドラインインターフェイスを開いて実行する：

pip install --upgrade ultralytics==8.0.186 wandb

インストール手順の詳細については、YOLOv8 インストールガイドを参照してください。問題が発生した場合、トラブルシューティングのヒントについては、「よくある問題」ガイドを参照してください。

Ultralytics YOLOv8 をWeights & Biases と統合する利点は？

Ultralytics YOLOv8 をWeights & Biases と統合すると、以下のような利点がある：

• リアルタイムのメトリクス追跡：トレーニング中の指標の変化を観察し、即座に洞察を得ることができます。
• ハイパーパラメータの最適化：学習率やバッチサイズなどを微調整することで、モデルのパフォーマンスを向上させます。
• 比較分析：異なるトレーニングの実行を並べて比較。
• リソース監視： CPU,GPU, メモリ使用量を追跡。
• モデル成果物の管理：モデルのチェックポイントに簡単にアクセスし、共有することができます。

これらの機能の詳細については、上記のWeights & Biases ダッシュボードのセクションを参照してください。

YOLOv8 トレーニング用にWeights & Biases を設定するにはどうすればよいですか？

YOLOv8 トレーニング用にWeights & Biases を設定するには、以下の手順に従う：

1. Weights & Biases を初期化するコマンドを実行する：

   import wandb
   wandb.login()
   

2. Weights & Biases ウェブサイトからAPIキーを取得します。
3. APIキーを使用して開発環境を認証します。

詳細な設定方法は、上記の「Weights & Biases の設定」セクションに記載されています。

Weights & Biases を使ってYOLOv8 モデルをトレーニングするには？

Weights & Biases を使用してYOLOv8 モデルをトレーニングするには、Python スクリプトで以下の手順を使用する：

import wandb
from wandb.integration.ultralytics import add_wandb_callback

from ultralytics import YOLO

# Initialize a Weights & Biases run
wandb.init(project="ultralytics", job_type="training")

# Load a YOLO model
model = YOLO("yolov8n.pt")

# Add W&B Callback for Ultralytics
add_wandb_callback(model, enable_model_checkpointing=True)

# Train and Fine-Tune the Model
model.train(project="ultralytics", data="coco8.yaml", epochs=5, imgsz=640)

# Validate the Model
model.val()

# Perform Inference and Log Results
model(["path/to/image1", "path/to/image2"])

# Finalize the W&B Run
wandb.finish()

このスクリプトは、Weights & Biases を初期化し、モデルをセットアップし、トレーニングし、結果をログに記録します。詳細については、上記の使用法のセクションをご覧ください。

なぜ他のプラットフォームではなく、Weights & Biases でUltralytics YOLOv8 を使うべきなのですか？

Ultralytics YOLOv8 Weights & Biases ：

• 高効率：トレーニング指標をリアルタイムで追跡し、パフォーマンスを最適化。
• 拡張性：堅牢なリソース監視および利用ツールにより、大規模なトレーニングジョブを容易に管理できます。
• インタラクティブ性：データの可視化とモデル管理のためのユーザーフレンドリーなインタラクティブUI。
• コミュニティとサポート：強力な統合ドキュメントとコミュニティサポート、柔軟なカスタマイズと拡張オプション。

Comet やClearML のような他のプラットフォームとの比較については、Ultralytics 統合 を参照。

コメント