标题: 用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%\.u2net\u2net.onnx

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