Tensori di immagini

Un'azienda di caffè ha un progetto di object detection in cui deve annotare gli oggetti di interesse, in questo caso gli shot di espresso. Hai creato un elenco con le coordinate del bounding box per un'immagine di un espresso. Ora devi convertire l'immagine e le coordinate in tensori.

torch e torchvision sono stati importati. torchvision.transforms è importato come transforms. L'immagine è stata caricata come image usando Image.open() dalla libreria PIL. Le coordinate del bounding box sono salvate nella variabile bbox.

espresso

Questo esercizio fa parte del corso

Deep Learning per Immagini con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

Converte bbox in tensori usando torch.tensor().
Rimodella bbox_tensor aggiungendo una dimensione batch con unsqueeze(0).
Crea una trasformazione per ridimensionare image a (224) e convertirla in un tensore immagine non scalato.
Applica transform a image.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Convert bbox into tensors
bbox_tensor = ____

# Add a new batch dimension
bbox_tensor = bbox_tensor.____

# Resize image and transform tensor
transform = transforms.Compose([
  transforms.____,
  transforms.____
])

# Apply transform to image
image_tensor = ____
print(image_tensor)

Modifica ed esegui il codice