用Python移除图片背景(抠图)

创建: 2023-03-14 10:19
https://scz.617.cn/python/202303141019.txt

宝玉在微博上分享了一个用Python移除图片背景的超级简单法子

https://m.weibo.cn/detail/4879137040501492

对图像处理七窍通了六窍，平日没有抠图需求，只当是多个技能留着，PS什么的敬而远之。用pip安装相应库之后，下列脚本指定输入、输出，傻瓜式抠图，测试无误。

#
# pip3 install rembg Pillow
# python3 GetPortrait.py input.jpg output.png
#
import sys, rembg, PIL
input   = PIL.Image.open( sys.argv[1] )
output  = rembg.remove( input )
output.save( sys.argv[2] )

初次执行GetPortrait.py时会自动下载这个文件，有175997641(176MB)这么大

https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx

会放到

$HOME/.u2net/u2net.onnx
%USERPROFILE%.u2netu2net.onnx

再次执行GetPortrait.py不会重复下载。既然是傻瓜式抠图，就存在一些问题，评论区有人问了，怎么识别谁是主体？我实测过程中发现，有些背景被识别进主体部分，没抠掉。

附一个对比效果，2003.8的红阳，二十年前的红阳