Usuwanie tła obrazu
W tym ćwiczeniu użyjesz tego samego zbioru danych flickr co poprzednio – zawiera on 30 000 obrazów wraz z opisami. Tym razem usuniesz tło obrazu.
Przykładowy obraz (image) i moduł potoku (pipeline) zostały już załadowane.
To ćwiczenie jest częścią kursu
Modele multimodalne z Hugging Face
Instrukcje do ćwiczenia
- Załaduj potok
image-segmentationz wstępnie wytrenowanym modelembriaai/RMBG-1.4i włączoną opcjątrust_remote_code. - Użyj potoku, aby przetworzyć wejściowy obraz
image.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Load the image-segmentation pipeline
pipe = ____
# Use the pipeline to predict the class of the sample image
outputs = ____
plt.imshow(outputs)
plt.show()