使用 Python 从图片中删除背景 - 无需再花费数小时在Photoshop上抠图

Rembg，一个允许您从图片中删除背景的 python 库。它非常容易安装和使用。您可以从命令行和 python 代码中使用它。它是用 python 编写的，可以在 GitHub 上找到，网址为https://github.com/danielgatis/rembg
它有 CPU 和 GPU 版本，所以如果你没有 gpu，你仍然可以使用这个库。

我将使用 Google Colab 来演示这个库的用法。（请先搭好梯子！）

（Google colab 允许我们在浏览器中运行代码，而无需在本地安装任何东西。）

（如果您不熟悉 colab 或不想运行整个代码，本文末尾有一个 Web 应用程序链接，您可以在其中直接上传图片并使用。）

让我们开始吧。

访问https://colab.research.google.com/并单击“新建笔记本”。
您将看到一个新的 colab notebook 打开。单击第一个单元格并输入以下命令并点击运行。这将安装“rembg”库和所有其他必需的库。

!pip install rembg

你应该看到这样的显示。

运行此步骤大约需要一分钟。

下一步是从给定的 url 下载图像，以便我们可以将其用作 rembg 的输入。以下代码会为您完成此操作。

from google.colab import files as FILE
import os
import requests
url="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-2.jpg"
img_data = requests.get(url).content
with open('input.jpg', 'wb') as handler:
    handler.write(img_data)

这会将图像下载到 google colab。现在使用以下命令行命令调用rembg 以从 content/input.jpg 中删除背景并将输出存储为 /content/image1_output_bg.jpg。

!rembg i  /content/input.jpg /content/image1_output_bg.jpg

这是我们使用的输入图像。

这是图像的 rembg 输出

这里还有几个例子。

但是我们如何在不运行代码的情况下使用我们自己的图片呢？
这是允许您在不运行任何代码的情况下进行拖放的链接。要使用它，只需上传您的图片并单击提交。
https://huggingface.co/spaces/KenjieDec/RemBG

最后，如果您有大量的图片需要删除背景，您也可以轻松实现。尽情享受 Rembg 给您带来的体验吧！