해당 코드는 rembg 공식 git으로부터 사용되었습니다.
아래의 version은 python 3.8 이상으로부터 변경되었으며, 본 post에서는 3.6버전에서의 구버전 rembg 코드를 참고하여 작성되었습니다.
- python 3.6
- flask>=1.1.2
- numpy>=1.19.5
- pillow>=8.0.1
- scikit-image>=0.17.2
- torch>=1.7.0
- torchvision>=0.8.1
- waitress>=1.4.4
- tqdm>=4.51.0
- requests>=2.24.0
- scipy>=1.5.4
- pymatting>=1.1.1
- filetype>=1.0.7
- hsh>=1.1.0
The install cmd is:
conda create -n your_prjname python=3.6
conda activate your_prjname
cd /생성된 가상환경경로
pip install -r rembg/requirements.txt
- your_prjname : 생성할 가상환경 이름 (파이썬버전은 3.6 고정)
아래의 링크를 통해 학습 weight 파일을 다운받습니다. 해당 파일은 rembg에서 학습한 pretrained file입니다. 해당 weight 파일은 "가상환경/rembg/src/rembg/cmd/" 에 위치하도록 합니다.
Example |
Remove the background from a remote image
python3 rembg/src/rembg/cmd/cli.py --input_path "원본 이미지가 저장된 로컬 경로" --output_path "배경제거 처리 된 이미지가 저장될 경로"
Original | Without background |
Original local folder | Without background local folder |