python 去除图片背景

使用工具 rembg

1. 安装

CPU版

pip install rembg

GPU版

pip install rembg[gpu]

2. 使用

1. 命令行调用

   1. 去除单张图片

      rembg i path/to/input.png path/to/output.png

   2. 去除多张图片

      rembg p path/to/input path/to/output

2. python中调用

    # -*- coding:utf-8 -*-
    # @Time   : 2022-05-15
    # @Author : carl_DJ

    from rembg.bg import  remove

    #待处理图片路径
    input_pic_path = input("请输入待处理图片地址：")

    #处理后图片保存路径
    output_pic_path = input("请输入保存地址：")

    #处理图片并保存
    with open(input_pic_path,'rb') as f_input:
        with open(output_pic_path,'wb') as f_output:
            input_pic = f_input.read()
            output_pic = remove(input_pic)
            f_output.write()
            #关闭
            f_output.close()
        f_input.close()

参考
https://blog.csdn.net/wuyoudeyuer/article/details/124782232