PaddleDetection

Intégrez W&B à PaddleDetection pour suivre l’entraînement de modèles de détection d’objets, consigner des métriques et visualiser les résultats.

PaddleDetection est un kit de développement de détection d’objets de bout en bout basé sur PaddlePaddle. Il détecte divers objets courants, segmente les instances, et suit et détecte les points clés à l’aide de modules configurables tels que les composants du réseau, les augmentations de données et les pertes.

PaddleDetection comprend désormais une intégration W&B native qui enregistre toutes vos métriques d’entraînement et de validation, ainsi que les points de contrôle de votre modèle et leurs métadonnées associées.

Le WandbLogger de PaddleDetection enregistre vos métriques d’entraînement et d’évaluation dans W&B, ainsi que les points de contrôle de votre modèle pendant l’entraînement.

Lire un article de blog de W&B qui montre comment intégrer un modèle YOLOX à PaddleDetection sur un sous-ensemble du jeu de données COCO2017.

Inscrivez-vous et créez une clé API

Une clé API authentifie votre machine auprès de W&B. Vous pouvez générer une clé API à partir de votre profil.

Pour une méthode plus directe, créez une clé API en accédant directement aux Paramètres utilisateur. Copiez immédiatement la clé API nouvellement créée et conservez-la dans un endroit sûr, par exemple dans un gestionnaire de mots de passe.

1. Cliquez sur l’icône de votre profil en haut à droite.
2. Sélectionnez Paramètres utilisateur, puis faites défiler jusqu’à la section Clés API.

Installez la bibliothèque wandb et connectez-vous

Pour installer la bibliothèque wandb en local et vous connecter :

Ligne de commande

1. Définissez la variable d’environnement WANDB_API_KEY avec votre clé API.

   export WANDB_API_KEY=<your_api_key>
   

2. Installez la bibliothèque wandb et connectez-vous.

   pip install wandb
   
   wandb login
   

Python

pip install wandb

import wandb
wandb.login()

Python notebook

!pip install wandb

import wandb
wandb.login()

Activez le WandbLogger dans votre script d’entraînement

Ligne de commande

Pour utiliser wandb via des arguments transmis à train.py dans PaddleDetection :

• Ajoutez l’option --use_wandb
• Les premiers arguments wandb doivent être précédés de -o (vous ne devez le passer qu’une seule fois)
• Chaque argument doit contenir le préfixe "wandb-" . Par exemple, tout argument transmis à wandb.init() recevra le préfixe wandb-

python tools/train.py 
    -c config.yml \ 
    --use_wandb \
    -o \ 
    wandb-project=MyDetector \
    wandb-entity=MyTeam \
    wandb-save_dir=./logs

`config.yml`

Ajoutez les arguments wandb au fichier config.yml sous la clé wandb :

wandb:
  project: MyProject
  entity: MyTeam
  save_dir: ./logs

Lorsque vous exécutez votre fichier train.py, un lien vers votre tableau de bord W&B est généré.

Retours ou problèmes

Si vous avez des retours ou rencontrez des problèmes avec l’intégration W&B, ouvrez une issue sur le PaddleDetection GitHub ou envoyez un e-mail à [e-mail].