YOLOX - CoreWeave Docs

YOLOX は、物体検出用のアンカーフリーな YOLO のバージョンです。YOLOX の W&B インテグレーションを使用すると、コマンドライン引数を 1 つ指定するだけで、トレーニング、検証、システムに関連するメトリクスをログする機能を有効にし、予測をインタラクティブに検証できます。このガイドでは、W&B で認証し、インテグレーションをインストールして、YOLOX の物体検出モデルをトレーニングする際に W&B ロギングを有効にする方法を説明します。これにより、W&B UI でメトリクスをトラッキングし、予測を確認できます。 APIキーは、お使いのマシンを W&B に対して認証するためのものです。APIキーはユーザープロフィールから発行できます。

より簡単な方法として、User Settings に移動して APIキーを作成することもできます。作成した APIキーはすぐにコピーし、パスワードマネージャーなどの安全な場所に保存してください。

右上のユーザープロフィールアイコンをクリックします。
User Settings を選択し、APIキー セクションまでスクロールします。

`wandb` ライブラリのインストールとログイン

wandb ライブラリをローカルにインストールしてログインするには、次の手順に従います。

Command line
Python
Python notebook

WANDB_API_KEY 環境変数に APIキーを設定します。<> で囲まれた値を自分の値に置き換えてください。
export WANDB_API_KEY=<your_api_key>
wandb ライブラリをインストールしてログインします。
pip install wandb wandb login

pip install wandb

import wandb
wandb.login()

!pip install wandb

import wandb
wandb.login()

メトリクスをログする

wandb ライブラリをインストールし、マシンの認証を済ませると、YOLOX トレーニングスクリプトから W&B ロギングを有効にできます。 --logger wandb コマンドライン引数を使用して、wandb によるログを有効にします。必要に応じて、wandb.init() が受け取るすべての引数も渡せます。各引数の先頭に wandb- を付けてください。 num_eval_imges は、モデル評価のために W&B が表にログする検証セットの画像と予測の数を制御します。コマンドを実行する前に、次のプレースホルダーを置き換えてください。

<project-name>: W&B のプロジェクト名。
<entity>: W&B の entity (ユーザー名またはチーム名) 。
<run-name>: このトレーニング run の名前。
<run-id>: この run の一意の ID。
<save-dir>: YOLOX がチェックポイントとログを保存するディレクトリ。
<num-images>: ログする検証画像の数。
<bool>: チェックポイントをログするかどうか (true または false) 。

# W&B にログインします
wandb login

# wandb ロガー引数を指定して YOLOX トレーニングスクリプトを呼び出します
python tools/train.py .... --logger wandb \
                wandb-project <project-name> \
                wandb-entity <entity> \
                wandb-name <run-name> \
                wandb-id <run-id> \
                wandb-save_dir <save-dir> \
                wandb-num_eval_imges <num-images> \
                wandb-log_checkpoints <bool>

例

トレーニング run が開始すると、YOLOX はトレーニング、検証、システムメトリクスを W&B プロジェクトにストリーミングします。そこで run を比較し、予測を確認できます。データが表示されたダッシュボードの例を以下に示します。 YOLOX のトレーニングと検証のメトリクスを表示するダッシュボードの例をご覧ください。

この W&B インテグレーションについて質問や問題がある場合は、YOLOX リポジトリで Issue を作成してください。

​サインアップして APIキーを作成する

​wandb ライブラリのインストールとログイン

​メトリクスをログする

​例

サインアップして APIキーを作成する

`wandb` ライブラリのインストールとログイン

メトリクスをログする

例