Малювання межувальної рамки

Підготувавши тензори зображення та межувальної рамки, тепер ви можете накреслити рамку поверх зображення й візуально перевірити її точність.

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()

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