Zacznij terazZacznij za darmo

Generator kotwic

Twój zespół tworzy modele detekcji obiektów oparte na architekturze Faster R-CNN z wykorzystaniem wstępnie wytrenowanych szkieletów. Twoim zadaniem jest stworzenie ramek kotwiczących (ang. anchor boxes), które posłużą jako ramki referencyjne do proponowania potencjalnych regionów zawierających obiekty.

Utworzysz 9 standardowych kotwic (3 rozmiary ramek i 3 współczynniki proporcji).

torch został już zaimportowany.

To ćwiczenie jest częścią kursu

Głębokie uczenie dla obrazów z PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

  • Zaimportuj AnchorGenerator z torchvision.models.detection.rpn.
  • Skonfiguruj rozmiary kotwic, podając 3 wartości: ((32, 64, 128),).
  • Skonfiguruj współczynniki proporcji, podając 3 wartości: ((0.5, 1.0, 2.0),).
  • Utwórz instancję AnchorGenerator, przekazując anchor_sizes i aspect_ratios.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# Import AnchorGenerator
____

# Configure anchor size
anchor_sizes = ____

# Configure aspect ratio
aspect_ratios = ____

# Instantiate AnchorGenerator
rpn_anchor_generator = ____
Edytuj i uruchom kod