Rembg,一個(gè)允許您從圖片中刪除背景的 Python/ target=_blank class=infotextkey>Python 庫。它非常容易安裝和使用。您可以從命令行和 python 代碼中使用它。它是用 python 編寫的,可以在 GitHub 上找到,網(wǎng)址為
https://github.com/danielgatis/rembg
它有 CPU 和 GPU 版本,所以如果你沒有 gpu,你仍然可以使用這個(gè)庫。

我將使用 google Colab 來演示這個(gè)庫的用法。(請先搭好梯子!)
(Google colab 允許我們在瀏覽器中運(yùn)行代碼,而無需在本地安裝任何東西。)

(如果您不熟悉 colab 或不想運(yùn)行整個(gè)代碼,本文末尾有一個(gè) Web 應(yīng)用程序鏈接,您可以在其中直接上傳圖片并使用。)
讓我們開始吧。
訪問
https://colab.research.google.com/并單擊“新建筆記本”。
您將看到一個(gè)新的 colab notebook 打開。單擊第一個(gè)單元格并輸入以下命令并點(diǎn)擊運(yùn)行。這將安裝“rembg”庫和所有其他必需的庫。
!pip install rembg
你應(yīng)該看到這樣的顯示。

運(yùn)行此步驟大約需要一分鐘。
下一步是從給定的 url 下載圖像,以便我們可以將其用作 rembg 的輸入。以下代碼會(huì)為您完成此操作。
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)
這會(huì)將圖像下載到 google colab。現(xiàn)在使用以下命令行命令調(diào)用rembg 以從 content/input.jpg 中刪除背景并將輸出存儲(chǔ)為
/content/image1_output_bg.jpg。
!rembg i /content/input.jpg /content/image1_output_bg.jpg

這是我們使用的輸入圖像。

這是圖像的 rembg 輸出

這里還有幾個(gè)例子。




但是我們?nèi)绾卧诓贿\(yùn)行代碼的情況下使用我們自己的圖片呢?
這是允許您在不運(yùn)行任何代碼的情況下進(jìn)行拖放的鏈接。要使用它,只需上傳您的圖片并單擊提交。
https://huggingface.co/spaces/KenjieDec/RemBG

最后,如果您有大量的圖片需要?jiǎng)h除背景,您也可以輕松實(shí)現(xiàn)。盡情享受 Rembg 給您帶來的體驗(yàn)吧!