ComeçarComece de graça

Gerador de âncoras

Sua equipe está desenvolvendo modelos de detecção de objetos baseados na arquitetura Faster R-CNN e usando backbones pré-treinados. Sua tarefa é criar caixas âncora que servirão como caixas delimitadoras de referência para propor possíveis regiões de objetos.

Você vai criar 9 âncoras padrão (3 tamanhos de caixa e 3 proporções de aspecto).

torch já foi importado para você.

Este exercício faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

Instruções do exercício

  • Importe AnchorGenerator de torchvision.models.detection.rpn.
  • Configure os tamanhos das âncoras com 3 valores: ((32, 64, 128),).
  • Configure a proporção de aspecto com 3 valores `((0.5, 1.0, 2.0),).
  • Instancie AnchorGenerator com anchor_sizes e aspect_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 = ____
Editar e executar o código