PaddleOCR

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

2 minute read

PaddleOCR は、多言語に対応した、素晴らしい、最先端の、そして実用的なOCRツールを開発し、PaddlePaddle で実装されたモデルのトレーニングをより良く行い、実際に活用できるようにすることを目的としています。PaddleOCR は、OCRに関連する様々な最先端のアルゴリズムをサポートし、産業ソリューションを開発しました。PaddleOCR には、トレーニングと評価メトリクスを、対応するメタデータと共にモデルのチェックポイントをログ記録するための Weights & Biases のインテグレーションが付属しています。

ブログと Colab の例

ICDAR2015 データセットで PaddleOCR を使用してモデルをトレーニングする方法については、こちら を参照してください。また、Google Colab もあり、対応するライブの W&B ダッシュボードは こちら から入手できます。このブログの中国語版もあります: W&B对您的OCR模型进行训练和调试

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

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

より効率的なアプローチとして、https://wandb.ai/authorize に直接アクセスして APIキーを生成できます。表示された APIキーをコピーして、パスワードマネージャーなどの安全な場所に保存してください。

右上隅にあるユーザープロファイルアイコンをクリックします。
[User Settings] を選択し、[API Keys] セクションまでスクロールします。
[Reveal] をクリックします。表示された APIキーをコピーします。APIキーを非表示にするには、ページをリロードします。

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

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

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()

`config.yml` ファイルに wandb を追加する

PaddleOCR では、yaml ファイルを使用して構成変数を指定する必要があります。構成 yaml ファイルの末尾に次のスニペットを追加すると、すべてのトレーニングと検証メトリクスが、モデルチェックポイントと共に W&B ダッシュボードに自動的に記録されます。

Global:
    use_wandb: True

wandb.init に渡したい追加のオプションの引数は、yaml ファイルの wandb ヘッダーの下に追加することもできます。

wandb:  
    project: CoolOCR  # (オプション) これは wandb プロジェクト名 です
    entity: my_team   # (オプション) wandb team を使用している場合は、ここに team 名を渡すことができます
    name: MyOCRModel  # (オプション) これは wandb run の名前です