Générateur d’ancres
Votre équipe développe des modèles de détection d’objets basés sur l’architecture Faster R-CNN et utilise des backbones pré-entraînés. Votre tâche consiste à créer des boîtes d’ancrage qui serviront de boîtes englobantes de référence pour proposer des régions d’objets potentielles.
Vous allez créer 9 ancres standard (3 tailles de boîtes et 3 rapports d’aspect).
torch a été importé pour vous.
Cet exercice fait partie du cours
Deep Learning pour l’image avec PyTorch
Instructions
- Importez
AnchorGeneratordepuistorchvision.models.detection.rpn. - Configurez les tailles d’ancrage avec 3 valeurs :
((32, 64, 128),). - Configurez le rapport d’aspect avec 3 valeurs
`((0.5, 1.0, 2.0),). - Instanciez
AnchorGeneratoravecanchor_sizesetaspect_ratios.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____