dsh-rembg:本地圖像背景移除插件

前言

DeepSeek Harness (DSH) 採用插件化架構,允許通過擴展功能來擴展能力。dsh-rembg 是一個本地圖像背景移除插件,它將 Python rembg 的功能集成到 DSH 生態中,爲智能體提供無需雲端 API 的背景處理能力。

插件簡介

dsh-rembg 是一個運行在 DSH 環境中的圖像處理插件,由開發者 mengruoa 維護,歸類於模型推理類別。它使用 MIT 許可證。該插件允許在 DSH 進程內執行背景移除操作,無需依賴系統 Python 環境。

安裝與啓用

安裝該插件需要 DSH 版本 >= 0.1.7-rc.1。

dsh plugin --profile web add dsh-rembg

安裝完成後,啓動 DSH 即可使用。

配置與初始化

插件安裝後,需要在 Web GUI 中進行初始化配置。

  1. 打開 Web GUI 側欄「插件」。
  2. 找到 dsh-rembg 插件,點擊進入。
  3. 在 rembg 行右側點擊配置入口。
  4. 選擇運行模式(CPU 或 NVIDIA GPU)。
  5. 點擊「初始化環境」以安裝必要的 Python 依賴。
  6. 在模型列表中選擇並安裝所需的背景移除模型。
  7. 保存設置。

配置說明:
- DSH 0.1.7-rc.1 起,配置入口已從「設置 → 插件 → 插件配置」遷移至插件頁的行配置。
- 設置值持久化保存在當前 profile 的 cordis.patch.yml 文件中該 rembg 行下。
- GPU 模式需要本地具備 NVIDIA GPU、相應的顯卡驅動,並能通過 nvidia-smi 命令檢測到。

使用方法

插件提供了 rembg 工具函數供模型調用。

函數參數:
- path:輸入圖片的絕對路徑,必填。
- model:已安裝的模型名稱,可選,若不指定則使用配置頁中設置的默認模型。

使用示例:

用 rembg 工具把 /path/to/photo.png 的背景去掉。

如需查看當前環境可用的模型列表,可先調用 rembg_models 函數,然後使用返回的有效模型調用 rembg。

注意事項

  • 該插件依賴和模型文件保存在插件安裝目錄中,不會修改系統 Python 環境。
  • 使用前請確保 DSH 進程有權限訪問目標圖片路徑及寫入輸出路徑。
  • GPU 模式對硬件和驅動有依賴,若環境不滿足,請切換至 CPU 模式。

結語

dsh-rembg 爲 DSH 生態提供了輕量級的本地圖像背景處理能力,適合需要在本地環境(特別是無公網 API 的場景)進行圖像處理的智能體開發。更多細節請參考插件目錄或源碼。

羽毛球分组比赛记分
小程序二维码

欢迎使用《羽毛球分组比赛记分》微信小程序

小夜