YOLOX

W&B を YOLOX と統合する方法。

YOLOX は、強力な オブジェクト検出 パフォーマンスを備えた YOLO のアンカーフリーバージョンです。 YOLOX の W&B インテグレーション を使用すると、トレーニング、検証、およびシステムに関連する メトリクス の ログ 記録を有効にでき、単一の コマンドライン 引数 で 予測 をインタラクティブに検証できます。

​

サインアップと APIキー の作成

APIキー は、W&B に対してマシンを認証します。 ユーザー プロフィールから APIキー を生成できます。

For a more streamlined approach, create an API key by going directly to User Settings. Copy the newly created API key immediately and save it in a secure location such as a password manager.

1. 右上隅にある ユーザー プロフィールアイコンをクリックします。
2. User Settings を選択し、 API Keys セクションまでスクロールします。

​

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

ローカルに wandb ライブラリをインストールしてログインするには：

Command Line

1. WANDB_API_KEY 環境変数 に APIキー を設定します。

   export WANDB_API_KEY=<your_api_key>
   

2. wandb ライブラリをインストールしてログインします。

   pip install wandb
   
   wandb login
   

Python

pip install wandb

import wandb
wandb.login()

Python notebook

!pip install wandb

import wandb
wandb.login()

​

メトリクス の ログ 記録

--logger wandb コマンドライン 引数 を使用して、wandb による ログ 記録を有効にします。 オプションとして、 wandb.init() が期待するすべての 引数 を渡すこともできます。 各 引数 の前に wandb- を付けてください。

num_eval_imges は、 モデルの評価 のために W&B Tables に ログ 記録される 検証セット の画像と 予測 の数を制御します。

# wandbにログイン
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>

​

例

YOLOX のトレーニングおよび検証メトリクスを表示したダッシュボードの例 ->

この W&B インテグレーション に関する質問や問題がありますか？ YOLOX リポジトリ で issue を作成してください。