withoutbg/withoutbg
Python SDK for local and cloud background removal (pip install withoutbg)
해결하는 문제
Python을 사용하여 이미지에서 배경을 제거하는 간단한 방법을 제공하며, 무료의 프라이빗한 로컬 모델을 실행하거나 고품질 클라우드 기반 API를 사용할 수 있는 선택권을 제공합니다.
작동 방식
이 라이브러리는 사용자가 두 가지 모드로 WithoutBG 객체를 인스턴스화할 수 있도록 합니다: 로컬 처리를 위한 open_weights()와 클라우드 처리를 위한 api()입니다. 로컬 모드는 DINOv3를 기반으로 하는 ONNX 모델(약 455 MB)을 다운로드하여 CPU에서 단 한 번의 패스로 깊이, 세그멘테이션, 매팅 및 정제 작업을 수행합니다. 클라우드 모드는 이미지를 원격 API로 전송하여 머리카락이나 털과 같은 복잡한 가장자리에서 더욱 뛰어난 결과를 제공합니다.
대상 사용자
자동화된 이미지 배경 제거가 필요한 스크립트, Jupyter notebook, 백엔드 또는 배치 처리 작업을 구축하는 개발자를 위해 설계되었습니다.
주요 특징
- 이중 모드 실행: 코드 한 줄로 로컬 오픈 웨이트(무료, 오프라인, 프라이빗)와 클라우드 API(고품질) 사이를 전환할 수 있습니다.
- 배치 처리: 모델 가중치를 다시 로드하지 않고 여러 이미지를 효율적으로 처리할 수 있는
remove_background_batch메서드가 포함되어 있습니다. - 로컬 CPU 지원: 로컬 모델은 ONNX를 통해 실행되므로 전용 GPU가 필요하지 않습니다.
- CLI 도구: Python 코드를 작성하지 않고도 빠르게 이미지를 처리할 수 있는 명령줄 인터페이스가 제공됩니다.
- PIL 통합: 이미지를 RGBA 모드의 PIL
Image객체로 반환하여 피사체를 새로운 배경에 합성하기가 매우 쉽습니다.