1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

cvičení

Tensory obrazów

Firma kawowa realizuje projekt wykrywania obiektów, w którym trzeba oznaczać interesujące obiekty – w tym przypadku porcje espresso. Masz listę ze współrzędnymi ramki ograniczającej dla obrazu z espresso. Teraz musisz przekonwertować obraz i współrzędne na tensory.

torch i torchvision są już zaimportowane. torchvision.transforms jest zaimportowane jako transforms. Obraz został wczytany jako image za pomocą Image.open() z biblioteki PIL. Współrzędne ramki ograniczającej są przechowywane w zmiennej bbox.

espresso

Pokyny

100 XP
  • Przekonwertuj bbox na tensory za pomocą torch.tensor().
  • Zmień kształt bbox_tensor, dodając wymiar wsadowy (batch) za pomocą unsqueeze(0).
  • Utwórz transformację, która zmieni rozmiar image do (224) i przekształci go w nieskalowany tensor obrazu.
  • Zastosuj transform do image.