YOLOX

YOLOX は、物体検出用のアンカーフリーな YOLO のバージョンです。YOLOX の Weights & Biases インテグレーションを使用すると、コマンドライン引数を 1 つ指定するだけで、トレーニング、検証、システムに関連するメトリクスをログする機能を有効にし、予測をインタラクティブに検証できます。このガイドでは、Weights & Biases で認証し、インテグレーションをインストールして、YOLOX の物体検出モデルをトレーニングする際に Weights & Biases ロギングを有効にする方法を説明します。これにより、W&B UI でメトリクスをトラッキングし、予測を確認できます。 APIキーは、お使いのマシンを Weights & Biases に対して認証するためのものです。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 トレーニングスクリプトから Weights & Biases ロギングを有効にできます。 --logger wandb コマンドライン引数を使用して、wandb によるログを有効にします。必要に応じて、wandb.init() が受け取るすべての引数も渡せます。各引数の先頭に wandb- を付けてください。 num_eval_imges は、モデル評価のために Weights & Biases が表にログする検証セットの画像と予測の数を制御します。コマンドを実行する前に、次のプレースホルダーを置き換えてください。

<project-name>: Weights & Biases のプロジェクト名。
<entity>: Forge のチーム名。
<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 はトレーニング、検証、システムメトリクスを Weights & Biases プロジェクトにストリーミングします。そこで run を比較し、予測を確認できます。データが表示されたダッシュボードの例を以下に示します。 YOLOX のトレーニングと検証のメトリクスを表示するダッシュボードの例をご覧ください。

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

Weights & Biases

ガイド

インテグレーション

リファレンス

サポート

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

メトリクスをログする

例

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

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

​メトリクスをログする

​例

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

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

メトリクスをログする

例