Obrázkové tensory

Kávová společnost pracuje na projektu detekce objektů, kde je potřeba anotovat sledované objekty – v tomto případě espressa. Vytvořil/a sis seznam se souřadnicemi ohraničujícího rámečku (bounding boxu) pro snímek espressa. Teď je potřeba převést obrázek i souřadnice na tensory.

torch a torchvision jsou naimportované. torchvision.transforms je naimportovaný jako transforms. Obrázek byl načten jako image pomocí Image.open() z knihovny PIL. Souřadnice ohraničujícího rámečku jsou uloženy v proměnné bbox.

espresso

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

Převeď bbox na tensor pomocí torch.tensor().
Přetvaruj bbox_tensor přidáním dávkové dimenze pomocí unsqueeze(0).
Vytvoř transformaci, která změní velikost obrázku image na (224) a převede ho na neškálovaný obrázkový tensor.
Aplikuj transform na image.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

Upravit a spustit kód