YOLOX - Weights & Biases Documentation

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 Keys セクションまでスクロールします。

`wandb` ライブラリをインストールしてログインする

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

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

<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 logger 引数を指定して 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 のプロジェクトに送信します。そこで Runs を比較し、予測を確認できます。データが表示されたダッシュボードがどのように見えるかは、次の例をご覧ください。 YOLOX のトレーニングおよび検証メトリクスを含むダッシュボードの例を参照してください。

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

​サインアップしてAPIキーを発行する

​wandb ライブラリをインストールしてログインする

​メトリクスをログする

​例

サインアップしてAPIキーを発行する

`wandb` ライブラリをインストールしてログインする

メトリクスをログする

例