Предсказание ограничивающих рамок

Вы обучили модель распознавания объектов. Теперь с её помощью нужно сформировать ограничивающие рамки и классификации для тестового изображения.

Модель доступна как model и уже переведена в режим оценки. Переменная test_image также доступна; библиотеки torch и torch.nn импортированы как nn.

Это упражнение является частью курса

Глубокое обучение для работы с изображениями на PyTorch

Посмотреть курс

Инструкции к упражнению

Получите выходные данные модели для test_image и сохраните их в переменную output.
Извлеките предсказанные ограничивающие рамки из выходных данных и сохраните их в переменную boxes.
Извлеките предсказанные оценки уверенности из выходных данных и сохраните их в переменную scores.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Get model's prediction
with torch.no_grad():
    output = ____

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

Редактировать и запускать код