NO.035-三行代码一秒抠图——By.杭州组

作者：芬达
关键词：抠图，rembg

一.背景

最近大促期间有很多小伙伴需要对商品主图进行处理，但是人工处理成本太高，需要在ps中处理这些图片，当然影刀也自带了很多ps指令，我想分享下在不使用ps的情况下如何一键抠图。

二.解决思路

使用了rembg模块对图像进行处理

rembg介绍：Rembg是一款开源的图像背景去除工具，它利用深度学习算法自动识别和分离图像中的前景和背景。通过训练神经网络，Rembg能够准确地提取出图像中的前景对象，并生成一个透明的前景图像，使您能够轻松地将其与其他背景图像合并。Rembg支持多种常见的图像格式，如PNG和JPEG，并且可以在Windows、Linux和macOS等多个平台上运行。无论是需要在设计、广告还是其他领域中进行图像处理，Rembg都能提供便捷、高效的背景去除功能。

现在来看下具体的使用方法：

1.安装rembg模块：

2.新建python模块，插入以下代码：

from rembg import remove
import cv2
from PIL import Image

def clear_background(input_path, output_path):

    input = Image.open(input_path)
    print("正在处理图像-{}!".format(input_path.split("\\")[-1]))

    output = remove(input)

    output.save(output_path)

    print("=" * 100)

打开图像--用rembg包中的remove方法删除背景图像--保存处理之后的结果

3.调用此模块并传入文件路径和保存路径：

4.结果展示：

使用前：

使用后：

耗时：

可以看到速度还是非常快的。

三.注意事项

1.用32位影刀可能会报错

2.需要提前把格式设置成png（可以用影刀自带的转换图片格式指令）

3.第一次运行需要安装一个模型的（一般都是使用u2net 这个通用的预训练模型，然后有其他需求的可以去安装其他的模型），会导致第一次运行的时候比较久。

4.感兴趣的小伙伴可以查看此链接~