AI 去除背景

- 1 min read

AI 去除背景

  • 使用 RMBG 库，基于 ONNX 模型，实现 AI 去除背景。

  • 模型下载地址 model.onnx

use rmbg::Rmbg;

fn main() -> anyhow::Result<()> {
    // 加载模型
    let rmbg = Rmbg::new("./model.onnx")?;

    // 读取原始图像
    let original_img = image::open("./image.png")?;

    // 移除背景
    let img_without_bg = rmbg.remove_background(&original_img)?;

    // 保存结果
    img_without_bg.save("./out.png")?;
    Ok(())
}

去除前后对比

  • 原始图像 rmbg

  • 去除背景后 rmbg