Тензори зображень

У кавової компанії є проєкт з розпізнавання об'єктів, де потрібно розмічати об'єкти інтересу — у цьому випадку шоти еспресо. Ви створили список із координатами обмежувальної рамки для зображення шоту еспресо. Тепер потрібно перетворити зображення та координати на тензори.

torch і torchvision вже імпортовано. torchvision.transforms імпортовано як transforms. Зображення завантажено як image за допомогою Image.open() з бібліотеки PIL. Координати обмежувальної рамки збережено у змінній bbox.

espresso

Ця вправа є частиною курсу

Глибоке навчання для зображень із PyTorch

Переглянути курс

Інструкції до вправи

Перетворіть bbox на тензор за допомогою torch.tensor().
Змініть форму bbox_tensor, додавши батч-вимір за допомогою unsqueeze(0).
Створіть перетворення, щоб змінити розмір image до (224) і перетворити на немасштабований тензор зображення.
Застосуйте transform до image.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

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

Редагувати та запускати код