CommencezCommencez gratuitement

Retrait de l'arrière-plan d'une image

Dans cet exercice, vous utiliserez le même jeu de données Flickr qu'auparavant, qui contient 30 000 images et leurs légendes associées. Cette fois, vous allez retirer l'arrière‑plan de l'image.

Photo de 2 personnes, dont 1 joue de la guitare

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

Cette activité fait partie du cours

Modèles multimodaux avec Hugging Face

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