Repolode
ZhengPeng7

ZhengPeng7/BiRefNet

고해상도 이진 이미지 분할과 배경 제거에 널리 쓰이는 딥러닝 모델 BiRefNet의 공식 PyTorch 구현입니다.

대체할 수 있는 제품:Remove.bgAdobe Photoshop Remove Background

스타
4,309
포크
344
열린 이슈
33
마지막 커밋
3일 전

어떤 프로젝트인가요?

BiRefNet은 고해상도에서 전경 객체와 배경을 분리하는 이진 이미지 분할용 딥러닝 모델입니다. 이 프로젝트는 2024년 CAAI AIR 논문의 공식 구현이며, PyTorch 기반의 학습, 테스트, 평가용 Python 스크립트를 제공합니다.

저장소에는 범용, 인물 및 일반 매팅, DIS, HRSOD, COD 등 여러 작업을 위한 사전 학습 가중치 모델 모음이 있습니다. Hugging Face Transformers를 통해 한 줄로 모델을 불러올 수 있으며, ONNX 변환과 FP16 추론 방법도 다룹니다. A100, RTX 4090, V100 GPU에서의 실행 시간과 메모리 사용량도 공개되어 있습니다.

또한 이미지와 정답 마스크를 폴더에 배치하는 방식으로 사용자 지정 데이터셋에 파인튜닝하는 방법을 안내합니다. ComfyUI 노드, Stable Diffusion WebUI 확장, TensorRT 및 GGUF 변환 등 커뮤니티 통합도 여러 가지 제공됩니다.

핵심 기능

  • 01이진 이미지 분할 작업에서 고해상도로 전경 객체를 분할합니다.
  • 02범용, 매팅, 인물 매팅, DIS, HRSOD, COD용 사전 학습 가중치를 제공합니다.
  • 03Hugging Face Transformers를 통해 한 줄로 모델을 불러옵니다.
  • 04ONNX 변환과 FP16 추론을 지원하며, 속도와 메모리 사용량 수치를 공개합니다.
  • 05학습 및 평가 스크립트를 사용해 사용자 지정 데이터셋으로 파인튜닝할 수 있습니다.

활용 사례

  • 제품 사진이나 인물 사진의 배경을 고해상도로 제거합니다.
  • PyTorch 또는 ONNX 모델을 기반으로 이미지·동영상 배경 제거 서비스를 구축합니다.
  • 특정 분야의 데이터셋에 맞춰 분할 모델을 파인튜닝합니다.
  • ComfyUI 또는 Stable Diffusion 워크플로에 배경 제거 기능을 노드로 추가합니다.

빠른 시작

  1. 1

    transformers를 사용해 Hugging Face에서 BiRefNet 모델을 불러옵니다

    birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)

사용 전 알아둘 점

  • README의 속도 수치는 RTX 4090에서 측정되었으므로, 고해상도 이미지를 실용적으로 처리하려면 일반적으로 GPU가 필요합니다.
  • 프로젝트 README에 따르면 모델을 더 발전시키려면 GPU 리소스가 더 필요하므로, 일부 고급 변형은 외부 컴퓨팅 리소스에 의존합니다.

토픽

자주 묻는 질문

BiRefNet은(는) 무엇인가요?
고해상도 이진 이미지 분할과 배경 제거에 널리 쓰이는 딥러닝 모델 BiRefNet의 공식 PyTorch 구현입니다. BiRefNet은 고해상도에서 전경 객체와 배경을 분리하는 이진 이미지 분할용 딥러닝 모델입니다. 이 프로젝트는 2024년 CAAI AIR 논문의 공식 구현이며, PyTorch 기반의 학습, 테스트, 평가용 Python 스크립트를 제공합니다.
BiRefNet은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
BiRefNet은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
BiRefNet은(는) 아직 유지보수되고 있나요?
활발히 유지보수 중. BiRefNet의 마지막 커밋은 2026년 10월 17일입니다.
BiRefNet은(는) 어디에 사용할 수 있나요?
제품 사진이나 인물 사진의 배경을 고해상도로 제거합니다. PyTorch 또는 ONNX 모델을 기반으로 이미지·동영상 배경 제거 서비스를 구축합니다. 특정 분야의 데이터셋에 맞춰 분할 모델을 파인튜닝합니다. ComfyUI 또는 Stable Diffusion 워크플로에 배경 제거 기능을 노드로 추가합니다.
BiRefNet는 어떻게 설치하나요?
transformers를 사용해 Hugging Face에서 BiRefNet 모델을 불러옵니다: birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)
BiRefNet은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/ZhengPeng7/BiRefNet

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트