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

Image background removal

In this exercise, you will use the same flickr dataset as previously, which has 30,000 images and associated captions. Now you will remove the image background.

Photo of 2 people, 1 is playing the guitar

The sample image (image) and pipeline module (pipeline) have been loaded.

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

Multi-Modal Models with Hugging Face

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

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

  • Load the image-segmentation pipeline with the briaai/RMBG-1.4 pretrained model, and trust_remote_code enabled.
  • Use the pipeline to modify the input image.

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

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

# Load the image-segmentation pipeline
pipe = ____

# Use the pipeline to predict the class of the sample image
outputs = ____

plt.imshow(outputs)
plt.show()
Редактировать и запускать код