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.

The sample image (image) and pipeline module (pipeline) have been loaded.
Este exercício faz parte do curso
Multi-Modal Models with Hugging Face
Instruções do exercício
- Load the
image-segmentationpipeline with thebriaai/RMBG-1.4pretrained model, andtrust_remote_codeenabled. - Use the pipeline to modify the input
image.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Load the image-segmentation pipeline
pipe = ____
# Use the pipeline to predict the class of the sample image
outputs = ____
plt.imshow(outputs)
plt.show()