コンテンツへスキップ

YOLO11実験のトラッキングとビジュアライゼーションの強化Weights & Biases

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



見るんだ: Ultralytics YOLO11の使い方Weights and Biases

このガイドでは、Ultralytics YOLO11とWeights & Biases との統合により、実験の追跡、モデルのチェックポイント、モデル性能の可視化が強化されたことを紹介します。また、統合の設定、トレーニング、微調整、Weights & Biases' インタラクティブ機能を使った結果の可視化についての説明も含まれています。

Weights & Biases

Weights & Biases 概要

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

YOLO11トレーニングWeights & Biases

Weights & Biases 、YOLO11のトレーニング・プロセスに効率化と自動化をもたらすことができます。

インストール

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

インストール

# Install the required packages for Ultralytics YOLO and Weights & Biases
pip install -U ultralytics wandb

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

設定Weights & Biases

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

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

SDKの初期セットアップ

import wandb

# Initialize your Weights & Biases environment
wandb.login(key="<API_KEY>")
# Initialize your Weights & Biases environment
wandb login <API_KEY>

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

使い方:YOLO11のトレーニングWeights & Biases

Weights & Biases 、YOLO11モデル・トレーニングの使い方の説明に入る前に、 Ultralytics 。これは、あなたのプロジェクト要件に最も適したモデルを選択するのに役立ちます。

使い方:YOLO11のトレーニングWeights & Biases

from ultralytics import YOLO

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

# Train and Fine-Tune the Model
model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo11n")
# Train a YOLO11 model with Weights & Biases
yolo train data=coco8.yaml epochs=5 project=ultralytics name=yolo11n

W&Bの主張

議論デフォルト説明
プロジェクトNoneローカルおよびW&Bに記録されるプロジェクト名を指定します。これにより、複数の実行をグループ化することができます。
名称Noneトレーニング実行の名前。これにより、サブフォルダの作成に使用される名前とW&Bロギングに使用される名前が決まります。

有効または無効Weights & Biases

Weights & Biases のロギングを有効または無効にしたい場合は、次のようにします。 wandb コマンドを使用する。デフォルトでは、Weights & Biases ロギングが有効になっている。

# Enable Weights & Biases logging
wandb enabled

# Disable Weights & Biases logging
wandb disabled

アウトプットを理解する

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

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

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

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

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

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

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

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

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

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

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

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

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

概要

このガイドでは、Ultralytics YOLO とWeights & Biases との統合について説明しました。この統合により、モデルのトレーニングおよび予測結果を効率的に追跡し、視覚化できることを説明しています。

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

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

よくあるご質問

Weights & Biases をUltralytics YOLO11 と統合するには?

Weights & Biases 、Ultralytics YOLO11と統合する:

  1. 必要なパッケージをインストールする:
pip install -U ultralytics wandb
  1. Weights & Biases アカウントにログインしてください:
import wandb

wandb.login(key="<API_KEY>")
  1. W&Bロギングを有効にしてYOLO11モデルをトレーニングする:
from ultralytics import YOLO

model = YOLO("yolo11n.pt")
model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo11n")

これにより、メトリクス、ハイパーパラメータ、およびモデルの成果物が、W&Bプロジェクトに自動的に記録されます。

Weights & Biases 、YOLO11との統合の主な特徴は?

主な特徴は以下の通り:

  • トレーニング中のリアルタイムのメトリクス追跡
  • ハイパーパラメータ最適化ツール
  • さまざまなトレーニング走行の比較分析
  • トレーニングの進捗状況をグラフで可視化
  • リソース監視 (CPU,GPU, メモリ使用量)
  • モデル成果物の管理と共有
  • 画像オーバーレイによる推論結果の表示

これらの機能は、実験の追跡、モデルの最適化、YOLO11プロジェクトでのより効果的な共同作業に役立つ。

YOLO11トレーニングのWeights & Biases ダッシュボードを見るにはどうしたらいいですか?

W&Bを統合したトレーニングスクリプトを実行した後:

  1. W&Bダッシュボードへのリンクがコンソール出力に提供されます。
  2. リンクをクリックするか、wandb.aiにアクセスしてアカウントにログインしてください。
  3. プロジェクトに移動して、詳細なメトリクス、ビジュアライゼーション、モデル・パフォーマンス・データを表示します。

ダッシュボードは、モデルのトレーニングプロセスに関する洞察を提供し、YOLO11モデルを効果的に分析・改善することを可能にします。

YOLO11のトレーニングで、Weights & Biases ロギングを無効にできますか?

はい、以下のコマンドを使ってW&Bロギングを無効にすることができます:

wandb disabled

ロギングを再度有効にするには

wandb enabled

これにより、トレーニングスクリプトを変更することなく、W&Bロギングを使用するタイミングを制御することができます。

Weights & Biases 、YOLO11モデルの最適化にどのように役立つのか?

Weights & Biases YOLO11のモデルを最適化するのに役立つ:

  1. トレーニングメトリクスの詳細なビジュアライゼーションの提供
  2. 異なるモデル間の比較が容易
  3. ハイパーパラメータ・チューニングのためのツールを提供
  4. モデル性能の共同分析を可能にする
  5. モデル成果物および結果の容易な共有の促進

これらの機能は、研究者や開発者がYOLO11モデルを改善するために、より速く反復し、データに基づいた決定を下すのに役立つ。

📅作成 10ヶ月前 ✏️更新しました 1ヶ月前

コメント