Gerador de âncora
A sua equipe tá desenvolvendo modelos de detecção de objetos com base na arquitetura Faster R-CNN e usando backbones pré-treinados. A sua tarefa é criar caixas âncora que vão servir como caixas delimitadoras de referência para sugerir possíveis regiões de objetos.
Você vai criar 9 âncoras padrão (3 tamanhos de caixa e 3 proporções).
torch foi importado para você.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Importe
AnchorGeneratordetorchvision.models.detection.rpn. - Configure os tamanhos das âncoras com 3 valores:
((32, 64, 128),). - Configure a proporção com 3 valores ``((0,5, 1,0, 2,0),)`.
- Instancie
AnchorGeneratorcomanchor_sizeseaspect_ratios.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____