Dessiner un cadre de sélection

Une fois l'image et les tenseurs de boîte englobante préparés, vous pouvez maintenant dessiner la boîte sur l'image et vérifier visuellement sa précision.

torch``torchvision ettorchvision.transforms ont été importés. L'image a déjà été transformée en tenseurs comme suit : image_tensor. Les coordonnées ont été attribuées aux variables suivantes : x_min, y_min, x_max, y_max.

Cet exercice fait partie du cours

Deep learning pour les images avec PyTorch

Afficher le cours

Instructions

Importez draw_bounding_boxes à partir de torchvision.utils.
Définissez l'bbox e du cadre englobant sous forme de liste composée de x_min, y_min, x_max et y_max.
Transmettez image_tensor et bbox_tensor à draw_bounding_boxes afin de tracer le cadre sur l'image et attribuez le résultat à img_bbox.
Convertissez le tenseur d'img_bbox s en image.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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

Modifier et exécuter le code