Générateur d'ancrages
Votre équipe met au point des modèles de détection d'objets basés sur l'architecture Faster R-CNN et utilisant des épines dorsales préentraînées. Votre tâche est de créer des boîtes d'ancrage servant de boîtes englobantes de référence pour proposer des régions d'objets potentielles.
Vous allez créer 9 ancrages standards (3 tailles de boîtes et 3 rapports d'aspect).
torch a été importé pour vous.
Cette activité fait partie du cours
Deep Learning pour les images avec PyTorch
Instructions de l’exercice
- 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 ce code d’exemple.
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____