mengruoa/dsh-rembg ↗★ 0

dsh-rembg

提供 CPU/GPU 图像背景移除工具,并带依赖与模型安装的设置界面。 适合需在 DSH 中去除图片背景的用户,GPU 模式需 NVIDIA 显卡与驱动。

套件
dsh-rembg
相容性
待驗證
Harness 依賴範圍
*
版本
0.1.15
授權
MIT
最近更新
2026年9月1日

安裝

$npx -p @deepseek-ai/dsh dsh plugin --profile web add github:mengruoa/dsh-rembg

配置

打开 Web GUI 的设置 → 插件 → 插件配置 → rembg 图像背景移除:

  1. 选择是否使用 GPU。
  2. 点击初始化环境安装对应的 Python 依赖。
  3. 在模型列表中安装要使用的模型。
  4. 选择默认模型并保存设置。

GPU 模式需要 NVIDIA GPU、可用驱动和 nvidia-smi;CPU 模式无需 NVIDIA GPU。依赖和模型保存在插件安装目录中,不会修改系统 Python。

使用

在 DSH 中让模型调用:

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

工具参数:

rembg(path, model?)
  • path:输入图片的绝对路径,必填。
  • model:已安装的模型名称,可选,默认使用设置中的模型。

如需查看可用模型,先调用 rembg_models,再使用已安装且校验有效的模型调用 rembg。处理完成后会返回输出文件信息。