如何解决如何在 tensorflow.js 上加载 onnx 模型?
我正在使用 tensorflow.js 创建一个程序。 它应该接收一个 onnx 文件并且能够使用 tf 加载它,从而能够进行推断。
我的问题是如何将它从 onnx 转换为 tfjs?
我宁愿只使用 js 来解决它。 (所以我不能使用 https://github.com/onnx/onnx-tensorflow 导致 python)
如果文件包含图像分类 cnn,resnet18、resnet34 或 mobilenet 有帮助。
解决方法
另一种解决方案可能是使用 ONNX Runtime Web 的 Javascript API - https://github.com/microsoft/onnxruntime/blob/master/js/web
ORT Web 直接采用 ONNX 模型。
旁注 - 通过 Tensorflow-ONNX 转换器 (https://github.com/onnx/tensorflow-onnx) 进行的 tfjs 模型转换也在进行中,即将发布
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。