Générateur d'ancrage
Votre équipe développe des modèles de détection d'objets basés sur l'architecture Faster R-CNN et utilise des réseaux neuronaux 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 potentiels.
Vous allez créer 9 ancres standard (3 tailles de boîte et 3 rapports d'aspect).
torch a été importé pour vous.
Cet exercice fait partie du cours
Deep learning pour les images avec PyTorch
Instructions
- Importez
AnchorGeneratorà partir detorchvision.models.detection.rpn. - Configurez la taille des ancres à l'aide de 3 valeurs :
((32, 64, 128),). - Configurez le rapport d'aspect à l'aide de 3 valeurs « ((0,5, 1,0, 2,0),) ».
- Veuillez instancier
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 = ____