이미지 배경 제거
이 연습 문제에서는 앞에서 사용한 flickr 데이터셋을 다시 사용합니다. 이 데이터셋에는 30,000장의 이미지와 그에 대응하는 캡션이 있습니다. 이제 이미지 배경을 제거해 보겠습니다.
샘플 이미지(image)와 파이프라인 모듈(pipeline)이 로드되어 있습니다.
이 연습은 강의의 일부입니다
Hugging Face로 배우는 멀티모달 모델
연습 안내
briaai/RMBG-1.4사전 학습 모델과trust_remote_code가 활성화된image-segmentation파이프라인을 로드하세요.- 파이프라인을 사용해 입력
image를 수정하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Load the image-segmentation pipeline
pipe = ____
# Use the pipeline to predict the class of the sample image
outputs = ____
plt.imshow(outputs)
plt.show()