ÎncepețiÎncepe gratuit

Generator de ancore

Echipa ta dezvoltă modele de detecție a obiectelor bazate pe arhitectura Faster R-CNN, folosind backbone-uri pre-antrenate. Sarcina ta este să creezi cutii de ancoră care să servească drept casete de delimitare de referință pentru propunerea regiunilor potențiale ale obiectelor.

Vei crea 9 ancore standard (3 dimensiuni de cutie și 3 rapoarte de aspect).

torch a fost deja importat pentru tine.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

  • Importă AnchorGenerator din torchvision.models.detection.rpn.
  • Configurează dimensiunile de ancoră cu 3 valori: ((32, 64, 128),).
  • Configurează raportul de aspect cu 3 valori: ((0.5, 1.0, 2.0),).
  • Instanțiază AnchorGenerator cu anchor_sizes și aspect_ratios.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Import AnchorGenerator
____

# Configure anchor size
anchor_sizes = ____

# Configure aspect ratio
aspect_ratios = ____

# Instantiate AnchorGenerator
rpn_anchor_generator = ____
Editează și rulează codul