Генератор якорів
Ваша команда розробляє моделі виявлення об'єктів на основі архітектури Faster R-CNN і використовує попередньо натреновані бекбони. Ваше завдання — створити якорні бокси, які слугуватимуть еталонними обмежувальними прямокутниками для пропонування потенційних регіонів об'єктів.
Ви створите 9 стандартних якорів (3 розміри боксів і 3 співвідношення сторін).
torch вже імпортовано для вас.
Ця вправа є частиною курсу
Глибоке навчання для зображень із PyTorch
Інструкції до вправи
- Імпортуйте
AnchorGeneratorзtorchvision.models.detection.rpn. - Налаштуйте розміри якорів із 3 значеннями:
((32, 64, 128),). - Налаштуйте співвідношення сторін із 3 значеннями
`((0.5, 1.0, 2.0),). - Створіть екземпляр
AnchorGeneratorзanchor_sizesіaspect_ratios.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____