浏览器本地使用 AI 移除图片背景

AI工具探索

浏览器本地使用 AI 移除图片背景

无需上传,零隐私担忧,利用WebGPU和AI技术在浏览器中实现高效的图片背景移除

最近在学习 AI 相关的前端知识,看到 Transformers.js 的一个实例很感兴趣就把它做成了一个工具。

通过在 WebWorker 中使用 Transformers.js 调用 WebGPU 运行 RMBG-1.4 模型,可以在浏览器本地使用 AI 移除图片背景。在 M1 PRO 上处理一张 4K 图片只需要 500ms。

AI 移除图片背景工具界面展示

技术实现参考

相关的源码在以下仓库,如果想自己实现,可以参考它:

xenova/transformers.js/examples/remove-background-client

提醒:要调用 WebGPU,需要切换到 Transformers.js V3 版本。