[파이썬] 파이썬을 활용한 이미지 배경 제거 - PIL, rembg

이미지 배경 제거

---

파이썬을 활용한 이미지 배경 제거 - PIL, rembg

---

들어가며

---

이미지를 활용하여 학습 데이터셋을 생성할 때, 이미지 전처리가 필요합니다. 불필요한 노이즈가 많이 생긴 이미지는 배경 제거가 필수 과정입니다. 이 글에서는 파이썬으로 이미지의 배경을 제거하는 방법을 소개합니다.

---

먼저 필요한 패키지를 import합니다.

from rembg import remove
from PIL import Image

---

배경을 제거할 이미지의 input 경로와 output 경로를 input_path와 output_path에 저장합니다.

input_path = 'horse_asis.jpg'
output_path = 'output.png'

---

이미지 샘플은 픽사베이의 무료 이미지를 사용하겠습니다. 샘플은 다음과 같습니다. 샘플 이미지의 배경인 잔디와 구름, 하늘을 제거하겠습니다.

이미지 배경 제거 전

---

제거하기 전의 이미지를 asis에 저장합니다.

asis = Image.open(input_path)

---

remove로 배경을 제거한 후 tobe에 저장하고, 위에서 정의한 경로로 이미지 파일을 저장합니다.

tobe = remove(asis)
tobe.save(output_path)

---

배경을 제거한 결과는 다음과 같습니다.

이미지 배경 제거 후

---