实现真正的一键抠图去背景 (python图像切片抠图)

实现真正的一键抠图去背景,python图像切片抠图

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

实现真正的一键抠图去背景,python图像切片抠图

我将使用 Google Colab 来演示这个库的用法。 (请先搭好梯子!)

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

实现真正的一键抠图去背景,python图像切片抠图

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

让我们开始吧。

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

!pip install rembg

你应该看到这样的显示。

实现真正的一键抠图去背景,python图像切片抠图

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

下一步是 从给定的 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

实现真正的一键抠图去背景,python图像切片抠图

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

实现真正的一键抠图去背景,python图像切片抠图

这是图像的 rembg 输出

实现真正的一键抠图去背景,python图像切片抠图

这里还有几个例子。

实现真正的一键抠图去背景,python图像切片抠图

实现真正的一键抠图去背景,python图像切片抠图

实现真正的一键抠图去背景,python图像切片抠图

实现真正的一键抠图去背景,python图像切片抠图

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

实现真正的一键抠图去背景,python图像切片抠图

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