dsh-rembg
v0.1.15
Published
DSH rembg plugin with CPU/GPU background removal and setup UI
Readme
dsh-rembg
DSH 图像背景移除插件,支持选择 CPU 或本地 NVIDIA GPU 模式。
安装
npm(推荐)
dsh plugin --profile web add dsh-rembgGitHub
dsh plugin --profile web add github:mengruoa/dsh-rembg启动 DSH:
dsh --profile web配置
打开 Web GUI 的设置 → 插件 → 插件配置 → rembg 图像背景移除:
- 选择是否使用 GPU。
- 点击初始化环境安装对应的 Python 依赖。
- 在模型列表中安装要使用的模型。
- 选择默认模型并保存设置。
GPU 模式需要 NVIDIA GPU、可用驱动和 nvidia-smi;CPU 模式无需 NVIDIA GPU。依赖和模型保存在插件安装目录中,不会修改系统 Python。
使用
在 DSH 中让模型调用:
用 rembg 工具把 /path/to/photo.png 的背景去掉。工具参数:
rembg(path, model?)path:输入图片的绝对路径,必填。model:已安装的模型名称,可选,默认使用设置中的模型。
如需查看可用模型,先调用 rembg_models,再使用已安装且校验有效的模型调用 rembg。处理完成后会返回输出文件信息。
卸载
dsh plugin --profile web remove dsh-rembg