Dibujar un cuadro delimitador

Una vez preparados los tensores de la imagen y del cuadro delimitador, ya puedes dibujar el cuadro sobre la imagen e inspeccionar visualmente su precisión.

torchSe han importado los archivos « torchvision » y «torchvision.transforms ». La imagen ya se ha transformado en tensores como image_tensor. Las coordenadas se han asignado a las variables: x_min, y_min, x_max, y_max.

Este ejercicio forma parte del curso

Aprendizaje profundo para imágenes con PyTorch

Ver curso

Instrucciones del ejercicio

Importa draw_bounding_boxes desde torchvision.utils.
Define el cuadro delimitador bbox como una lista compuesta por x_min, y_min, x_max y y_max.
Pasa image_tensor y bbox_tensor a draw_bounding_boxes para dibujar el cuadro sobre la imagen y asigna la salida a img_bbox.
Convierte el tensor « img_bbox » en una imagen.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

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

Editar y ejecutar código