НачатьНачать бесплатно

Сегментация изображения с помощью маски

Теперь, когда бинарная маска готова, с её помощью можно выделить объект — кошку — из изображения.

Для этого загрузите исходное изображение и преобразуйте его в тензор. Затем создайте тензор объекта, применив маску к исходному изображению. В конце отобразите результат.

transforms из torchvision уже импортированы, а binary_mask, созданная в предыдущем упражнении, доступна вам.

Это упражнение является частью курса

Глубокое обучение для работы с изображениями на PyTorch

Посмотреть курс

Инструкции к упражнению

  • Создайте image_tensor, применив преобразование ToTensor(), заданное как transform, к исходному изображению.
  • Выполните сегментацию изображения, применив binary_mask к image_tensor, и присвойте результат переменной object_tensor.
  • Примените уже заданное преобразование to_pil_image к object_tensor, чтобы отобразить результат.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Load image and transform to tensor
image = Image.open("images/Egyptian_Mau_123.jpg")
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = ____

# Segment object out of the image
object_tensor = ____

# Convert segmented object to image and display
to_pil_image = ____
object_image = to_pil_image(object_tensor)
plt.imshow(object_image)
plt.show()
Редактировать и запускать код