파이썬과 IT 관련 지식을 정리하고 있는 잡학다식 개발자입니다.

오늘은 쇼핑몰을 운영하시거나 블로그 썸네일을 만드시는 분들, 혹은 이미지 전처리 자동화가 필요한 분들이 자주 겪는 '누끼 따기(배경 제거)'에 대해 이야기해 보려고 합니다.

이론과 현실의 간극: 왜 교과서적인 방법은 실무에서 실패할까?

구글에 '파이썬 이미지 배경 제거'를 검색하면 가장 먼저 나오는 교과서적인 방법은 보통 OpenCV를 이용한 색상 임계값(Thresholding)이나 윤곽선(Contour) 검출 방식입니다. 이론적으로는 배경과 피사체의 색상 차이를 이용해 분리해 내는 훌륭한 원리죠.

하지만 현실의 실무 데이터는 그렇게 만만하지 않습니다. 조명에 따라 생기는 그림자, 피사체와 비슷한 색상의 배경, 특히 '머리카락'이나 '퍼(Fur)' 같은 복잡한 경계선을 만나면 이 교과서적인 방법은 여지없이 무너집니다. 결국 파라미터 값을 조정하다가 밤을 새우는 이른바 '탁상공론'식 코딩이 되고 맙니다.

실제 현업의 해결책: AI 기반 rembg 라이브러리

그래서 실제 서비스 환경이나 실무에서는 픽셀의 색상을 계산하는 대신, U-Net 기반의 딥러닝 모델이 적용된 rembg라는 라이브러리를 사용합니다. 복잡한 수학적 연산이나 파라미터 튜닝 없이, AI가 알아서 피사체와 배경을 완벽하게 분리해 줍니다.

구형 방식인 requirements.txt 대신, 요즘 파이썬 생태계 표준인 pyproject.toml을 이용해 환경을 세팅하고 코드를 실행하는 방법을 공유합니다.

1. 환경 세팅 (pyproject.toml)

Ini, TOML

[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "background-removal-demo"
version = "0.1.0"
description = "rembg를 활용한 이미지 배경 제거 자동화 스크립트"
requires-python = ">=3.9"
dependencies = [
    "rembg>=2.0.50",
    "Pillow>=10.0.0"
]

2. 실행 코드 (remove_background.py)

Python

"""
이미지 배경 제거 자동화 스크립트
- rembg 라이브러리를 활용하여 딥러닝 모델 기반으로 피사체와 배경을 분리함
- 복잡한 윤곽선(머리카락, 그림자 등)을 처리하기 위한 실무 최적화 방식
"""
from rembg import remove
from PIL import Image
import os

def process_image(input_path: str, output_path: str):
    # 입력 및 출력 파일의 유효성 검증
    if not os.path.exists(input_path):
        print(f"[Error] 입력 파일을 찾을 수 없습니다: {input_path}")
        return

    try:
        # 이미지를 메모리에 로드
        input_image = Image.open(input_path)
        
        # rembg 라이브러리를 통한 배경 제거 처리 (추론)
        output_image = remove(input_image)
        
        # 투명도(Alpha 채널)를 보존하기 위해 PNG 포맷으로 저장
        output_image.save(output_path, format="PNG")
        print(f"[System] 배경 제거 완료. 결과물 저장 경로: {output_path}")

    except Exception as e:
        print(f"[Error] 배경 제거 처리 중 예외 발생: {e}")

if __name__ == "__main__":
    # 테스트용 파일 경로 지정 (실제 파일 경로로 수정하여 사용)
    INPUT_FILE = "sample_input.jpg"
    OUTPUT_FILE = "sample_output.png"
    
    process_image(INPUT_FILE, OUTPUT_FILE)

마치며

코드를 보시면 아시겠지만, 실제 현업에서 문제를 해결하는 방식은 때로는 교과서적인 기초 원리보다 이미 잘 만들어진 도구를 얼마나 적절하게 가져다 쓰느냐에 달려있기도 합니다.

위의 코드를 복사해서 직접 실행해 보시면 퀄리티에 꽤 놀라실 겁니다. 만약 텍스트와 코드만으로 이해가 어렵거나, 터미널 환경 세팅부터 실제 이미지가 깔끔하게 누끼 따지는 시각적인 과정 전체를 눈으로 확인하고 싶으신 분들은 아래에 제가 정리해 둔 영상 링크를 참고해 주셔도 좋습니다.

오늘도 오류 없는 평온한 코딩 하시길 바랍니다. 감사합니다.

https://youtu.be/HzuSu2b_5N4