1. Nauka
  2. /
  3. Kursy
  4. /
  5. Głębokie uczenie dla obrazów z PyTorch

Connected

ćwiczenie

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.

Instrukcje

100 XP
  • 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.