优化 ONNX ISNet 模型

作者: xenova创建于 2025年3月3日更新于 2025年3月3日

伟大的库!我使用 Netron 检查了模型,发现还有一些优化空间。使用名为 onnxslim 的工具,可以删除一大堆冗余节点(没有什么大问题,但可以节省一些加载时间)。这些优化模型的一个好处是,它们删除所有 Shape 操作,在 WebGPU 上运行 ONNX Runtime Web 时,这些操作必须在 CPU 上运行。因此,现在没有任何 Shape 操作了,这意味着 CPU 和 GPU 之间的数据传输减少了。我已经将优化模型上传到 这里。如果你愿意的话,也可以将它们适用到这个项目中。

内容来源: imgly/background-removal-js