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 的场景)进行图像处理的智能体开发。更多细节请参考插件目录或源码。

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

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

小夜