rembg를 활용하여 사진 배경 지우기

rembg 깃허브

rembg란?

rembg는 이미지에서 배경을 제거하는 데 사용되는 파이썬 라이브러리이다. 이미지 처리 작업, 특히 전자상거래 제품 사진, 인물 사진, 또는 배경을 제거하여 피사체를 돋보이게 하고 싶은 모든 이미지에 유용하게 사용할 수 있다.

rembg 사용하기

• 요구사항 python: >3.7, <3.13

1) 가상환경 만들기 및 라이브러리 설치

• anaconda powershell prompt에서

  conda create -n p38_rembg_test python=3.8
  

• vscocde rembg_test 프로젝트 터미널

  pip install rembg
  

2) 테스트 이미지 준비

키티 캐릭터 배경에 무늬가 많아서 키티만 인식을 할 수 있을까?

3) 소스코드

# rembg 패키지에서 remove 클래스 불러오기
from rembg import remove 

# PIL 패키지에서 Image 클래스 불러오기
from PIL import Image 

# hellokitty.png파일 불러오기
img = Image.open("img/hellokitty.png") 

# 배경 제거하기
out = remove(img) 

# 변경된 이미지 저장하기
out.save("hellokitty_removebg.png") 

4) 결과

완벽하게 배경을 삭제했다! 다만 얼굴이랑 핸드폰 사이의 여백은 삭제하지 못했지만 이정도면 성능이 엄청 좋다는 것을 알 수 있다.

---