CommencerCommencez gratuitement

Suppression de l’arrière-plan d’image

Dans cet exercice, vous allez utiliser le même jeu de données flickr que précédemment, qui contient 30 000 images et leurs légendes associées. Vous allez maintenant supprimer l’arrière-plan de l’image.

Photo de 2 personnes, l’une joue de la guitare

L’image d’exemple (image) et le module de pipeline (pipeline) ont été chargés.

Cet exercice fait partie du cours

<cours>Modèles multimodaux avec Hugging Face</cours>
Voir le cours

Instructions de l’exercice

  • Chargez le pipeline image-segmentation avec le modèle préentraîné briaai/RMBG-1.4, et activez trust_remote_code.
  • Utilisez le pipeline pour modifier l’image en entrée.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Load the image-segmentation pipeline
pipe = ____

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

plt.imshow(outputs)
plt.show()
Modifier et exécuter le code