python 간단하게 배경 색 변경하기

 

참고 포스트: https://towardsdatascience.com/change-the-background-of-any-image-with-5-lines-of-code-23a0ef10ce9a

Change the Background of Any Image with 5 Lines of Code

 

해당 원본 글을 참고해서 진행하였으며,

필요한 라이브러리 설치는 다음과 같습니다.

 

Install Tensorflow with:(PixelLib supports tensorflow 2.0 and above)

• pip3 install tensorflow
• pip3 install pixellib
• pip3 install pixellib — upgrade

 

여기서 'deeplabv3_xception_tf_dim_ordering_tf_kernels.h5' 파일을 모델로 불러와 사용하는데

해당 파일은 here 여기서 받아서 다운로드 합니다.

 

만약, 배경을 흰색(255, 255, 255)로 변경하고 싶은 경우에는 다음과 같이 하면 

결과 이미지 경로에 파일이 저장하는 것으로 확인하였습니다.

 

import pixellib
from pixellib.tune_bg import alter_bg

change_bg = alter_bg()
change_bg.load_pascalvoc_model("deeplabv3_xception_tf_dim_ordering_tf_kernels.h5")
change_bg.color_bg({원본 이미지 경로}, colors=(255, 255, 255), output_image_name={결과 이미지 경로})

 

원본: 

 

결과: