CommencerCommencer 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

Modèles multimodaux avec Hugging Face

Afficher le cours

Instructions

  • 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 cet exemple de code.

# 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