1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

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

Pokyny

100 XP
  • 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.