Desenhar uma caixa delimitadora

Depois de preparar a imagem e os tensores da caixa delimitadora, você pode desenhar a caixa em cima da imagem e conferir se tá tudo certo.

torch``torchvision etorchvision.transforms foram importados. A imagem já foi transformada em tensores como image_tensor. As coordenadas foram atribuídas às variáveis: x_min, y_min, x_max, y_max.

Este exercício faz parte do curso

Aprendizado profundo para imagens com PyTorch

Ver curso

Instruções do exercício

Importe draw_bounding_boxes de torchvision.utils.
Defina a caixa delimitadora bbox como uma lista composta por x_min, y_min, x_max e y_max.
Passa image_tensor e bbox_tensor para draw_bounding_boxes pra desenhar a caixa em cima da imagem e atribui a saída para img_bbox.
Converte o tensor “ img_bbox ” em imagem.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

# 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 e executar o código