Генератор якорей
Ваша команда разрабатывает модели обнаружения объектов на основе архитектуры 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 = ____