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.
Diese Übung ist Teil des Kurses
Multi-Modal Models with Hugging Face
Anleitung zur Übung
- Load the
image-segmentation
pipeline with thebriaai/RMBG-1.4
pretrained model, andtrust_remote_code
enabled. - Use the pipeline to modify the input
image
.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Load the image-segmentation pipeline
pipe = ____
# Use the pipeline to predict the class of the sample image
outputs = ____
plt.imshow(outputs)
plt.show()