利用 ONNX Runtime 在浏览器中实现 20 倍速的背景移除

2024-06-22

本文介绍了如何利用 ONNX Runtime 和 WebGPU 在浏览器中实现高效的图像背景移除。文章指出,传统的服务器端背景移除方式存在延迟高、成本高、隐私风险等问题,而浏览器端处理可以有效解决这些问题。文章详细介绍了 ONNX Runtime Web 的 WebGPU 执行提供程序,以及如何将 PyTorch 训练的模型转换为 ONNX 格式并在浏览器中运行。通过性能评估,文章展示了 WebGPU 相比 CPU 在背景移除任务上的显著性能优势,实现了近乎实时的处理速度,为浏览器成为真正的应用平台迈出了重要一步。

(img.ly)
44