ComfyUI-RMBG: 완벽한 피사체 추출, 배경 제거 및 정밀한 세그먼테이션 194

ComfyUI-RMBG란 무엇인가요?

ComfyUI-RMBG는 ai 기술 기반의 이미지 처리 도구로, 주로 배경 제거 및 객체 분할에 사용됩니다. RMBG-2.0, InSPYRENET, BEN 등 다양한 고급 모델을 통합하여 효율적이고 정밀하게 누끼 따기 작업을 수행할 수 있습니다.

ComfyUI-RMBG 핵심 기능:

배경 제거 (RMBG Node):

• RMBG - 2.0, INSPYRENET, BEN, BEN2 등 다양한 모델을 사용할 수 있습니다. 여러 가지 배경 옵션을 선택할 수 있으며, 일괄 처리도 지원합니다.

• RMBG - 2.0 모델은 BiRefNet 아키텍처를 기반으로 구축되어, 복잡한 환경에서 배경을 제거할 때 정밀도가 높고 가장자리의 세부 사항도 잘 보존됩니다.

객체 분할 (Segment Node):

• 텍스트 프롬프트를 통해 객체 감지를 수행할 수 있으며, 라벨 스타일 및 자연어 입력을 지원하고 SAM (Segment Anything Model)을 활용하여 고정밀 분할을 구현합니다.

또한, 이 프로젝트에서는 RMBG - 2.0, INSPYRENET, BEN, BEN2, BiRefNet, SAM, GroundingDINO 등 다양한 모델을 수동으로 다운로드하여 설치할 수 있습니다. 이러한 모델들은 각각 이미지 처리 시나리오에서 고유한 장점을 가지고 있습니다:

• RMBG - 2.0: BiRefNet 아키텍처를 기반으로 개발되었으며, 복잡한 장면의 배경 추출에 사용됩니다. 투명, 검정, 흰색 등 다양한 배경 옵션이 있으며, 가장자리 품질 최적화를 통해 보다 자연스러운 누끼 작업을 수행합니다.

• InSPYRENET: 주로 비디오 배경 제거를 목표로 하며, 일괄 처리 효율이 높고 임계값을 조정하여 배경 효과를 개선할 수 있습니다.

• BEN 모델: 작고 효율적이어서 컴퓨팅 성능이 낮은 장치에 적합하며, 정밀도가 비교적 높고 컴퓨팅 리소스 소모가 적습니다.

ComfyUI-RMBG 기술 요구 사항

• ComfyUI 및 Python 3.10+ 설치가 필요합니다.

• 필요한 Python 패키지에는 torch, torchvision, Pillow, numpy 등이 있습니다.

ComfyUI-RMBG 설치 방법

1. ComfyUI-Manager를 통한 설치

ComfyUI-Manager에서 ComfyUI-RMBG를 검색하여 설치합니다.설치가 완료되면 다음 명령을 실행하여 의존성을 설치합니다:

./ComfyUI/python_embeded/python -m pip install -r requirements.txt

2. 저장소 수동 복제

저장소를 ComfyUI의 custom_nodes 폴더에 복제합니다:

cd ComfyUI/custom_nodes
git clone https://github.com/1038lab/ComfyUI-RMBG

3. 의존성 설치.

4. 모델 수동 다운로드

• 처음 사용할 때 모델은 ComfyUI/models/RMBG/ 폴더에 자동으로 다운로드됩니다.

• 사용자가 모델 파일을 수동으로 다운로드하여 지정된 디렉터리에 배치할 수도 있습니다. 예를 들어:

• RMBG-2.0 모델은 /ComfyUI/models/RMBG/RMBG-2.0에 배치합니다.

• INSPYRENET 모델은 /ComfyUI/models/RMBG/INSPYRENET에 배치합니다.

• 기타 모델의 다운로드 링크와 저장 경로는 프로젝트 문서를 참고하실 수 있습니다.

ComfyUI-RMBG 사용 시나리오

• 전자상거래 제품 사진: 원 클릭으로 누끼를 따고, 복잡한 배경에서 제품을 추출하여 통일된 배경으로 교체합니다.

• 비디오 편집: 비디오 일괄 처리를 지원하여 배경을 빠르게 교체합니다.

• AI 의상 교체 및 애니메이션 제작: 다른 플러그인과 결합하여 배경 교체 및 캐릭터 의상 변경을 구현합니다.

• 예술 창작: 컴퓨팅 성능이 낮은 장치에서 BEN과 같은 경량 모델을 사용하여创意적인 표현에 집중합니다.

GitHub 저장소 주소:https://github.com/1038lab/ComfyUI-RMBG