Rembg,一個允許您從圖片中刪除背景的 Python/ target=_blank class=infotextkey>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 給您帶來的體驗吧!