标题: 用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

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不会重复下载。既然是傻瓜式抠图，就存在一些问题，评论区有人问了，怎么识别谁是主体？我实测过程中发现，有些背景被识别进主体部分，没抠掉。