前言¶
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 中進行初始化配置。
- 打開 Web GUI 側欄「插件」。
- 找到
dsh-rembg插件,點擊進入。 - 在
rembg行右側點擊配置入口。 - 選擇運行模式(CPU 或 NVIDIA GPU)。
- 點擊「初始化環境」以安裝必要的 Python 依賴。
- 在模型列表中選擇並安裝所需的背景移除模型。
- 保存設置。
配置說明:
- 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 的場景)進行圖像處理的智能體開發。更多細節請參考插件目錄或源碼。