Ankargenerator
Ditt team utvecklar objektdetekteringsmodeller baserade på Faster R-CNN-arkitekturen och använder förtränade ryggrader. Din uppgift är att skapa ankarboxar som fungerar som referensbegränsningsboxar för att föreslå potentiella objektregioner.
Du ska skapa 9 standardankare (3 boxstorlekar och 3 bildförhållanden).
torch har redan importerats åt dig.
Den här övningen är en del av kursen
Djupinlärning för bilder med PyTorch
Övningsinstruktioner
- Importera
AnchorGeneratorfråntorchvision.models.detection.rpn. - Konfigurera ankarstorlekarna med 3 värden:
((32, 64, 128),). - Konfigurera bildförhållandet med 3 värden:
((0.5, 1.0, 2.0),). - Instantiera
AnchorGeneratormedanchor_sizesochaspect_ratios.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____