PaddleDetection
PaddleDetection は PaddlePaddle に基づいたエンドツーエンドのオブジェクト検出開発キットです。ネットワークコンポーネント、データ拡張、損失などのモジュール設計で、多様な主流のオブジェクト検出、インスタンスセグメンテーション、トラッキングおよびキーポイント検出アルゴリズムを実装しています。
PaddleDetection は、トレーニングおよび検証メトリクス、モデルチェックポイント、それに対応するメタデータをログするためのW&Bインテグレーションが組み込まれています。
例: ブログとColab
PaddleDetectionを使ってCOCO2017データセットのサブセットでYOLOXモデルをトレーニングする方法については、こちらのブログをお読みください。また、Google Colabも利用可能で、対応するライブW&Bダッシュボードはこちらからアクセスできます。
PaddleDetection WandbLogger
PaddleDetection WandbLoggerは、トレーニングと評価のメトリクス、およびトレーニング中のモデルチェックポイントを Weights & Biases にログします。
Weights & BiasesとPaddleDetectionの利用
W&Bにサインアップとログイン
こちらからサインアップして、無料のWeights & Biasesアカウントを作成し、wandbライブラリをインストールしてください。ログインするには、www.wandb.aiであなたのアカウントにサインインしている必要があります。サインインすると、[**認証ページ**](https://wandb.ai/authorize)で**APIキーが見つかります**。
- Command Line
- Notebook
pip install wandb
wandb login
!pip install wandb
wandb.login()
トレーニングスクリプトでWandbLoggerを有効にする
CLIを使用する場合
PaddleDetectionの train.py に引数を渡してwandbを使用するには:
--use_wandbフラグを追加する- 最初のwandb引数は
-oに続ける（これを一度だけ渡せばok） - 個々のwandb引数には
wandb-プレフィックスを付ける。例えばwandb.initに渡す引数にはwandb-プレフィックスを付けます
python tools/train.py
-c config.yml \
--use_wandb \
-o \
wandb-project=MyDetector \
wandb-entity=MyTeam \
wandb-save_dir=./logs
config.ymlファイルを使用する場合
configファイルを介してwandbを有効にすることもできます。config.ymlファイルにwandb引数を下記のように追加します:
wandb:
project: MyProject
entity: MyTeam
save_dir: ./logs
Weights & Biasesを有効にして train.py ファイルを実行すると、W&Bダッシュボードへのリンクが生成されます:
フィードバックや問題点
Weights & Biasesインテグレーションに関するフィードバックや問題がある場合は、PaddleDetection GitHubで問題を報告するか、support@wandb.com にメールしてください。