Рисование ограничивающего прямоугольника

Теперь, когда тензоры изображения и ограничивающего прямоугольника готовы, можно нарисовать прямоугольник поверх изображения и визуально проверить точность разметки.

torch, torchvision и torchvision.transforms уже импортированы. Изображение преобразовано в тензор и сохранено в переменной image_tensor. Координаты заданы в переменных: x_min, y_min, x_max, y_max.

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

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

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

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

Импортируйте draw_bounding_boxes из torchvision.utils.
Определите ограничивающий прямоугольник bbox в виде списка из значений x_min, y_min, x_max и y_max.
Передайте image_tensor и bbox_tensor в функцию draw_bounding_boxes, чтобы нарисовать прямоугольник поверх изображения, и сохраните результат в переменную img_bbox.
Преобразуйте тензор img_bbox в изображение.

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

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

# Import draw_bounding_boxes
____

# Define the bounding box coordinates
bbox = ____
bbox_tensor = torch.tensor(bbox).unsqueeze(0)

# Implement draw_bounding_boxes
img_bbox = ____(____, ____, width=3, colors="red")

# Tranform tensors to image
transform = transforms.Compose([
    ____
])
plt.imshow(transform(img_bbox))
plt.show()

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