如何在浏览器上运行PyTorch模型?

6
我想知道在浏览器中运行目标检测模型的选项有哪些,目前我找到了以下几个选项:
  1. Streamlit:非常简单但需要服务器资源运行,包括自己的相机。此外,我认为它不可能作为一个网站实现。
  2. Brython:是JS的替代品吗?我找不到加载pytorch或库的方法。
  3. Onnx:似乎是一个非常好的选择,但我不太明白如何在JS中运行模型,因为我只使用过Python和PyTorch,请问其学习路线图是什么?与Webassembly有什么关系?
  4. NCNN:这是我了解最少的方法,它似乎能够在基本上任何设备上工作,包括Webassembly、Windows、Android等,但同样,我无法理解如何将我的Python-PyTorch模型用于这个框架,因为我不知道路线图。
还有其他选项吗?
2个回答

1

TensorFlow.js

不过,您仍需要使用类似ONNX之类的工具来转换您的模型。


0

更新一下。

Tensorflow.js适合于TensorFlow模型,而onnx runtime web则是其他模型的好选择。需要做的事情如下:

  • 将您的模型转换为onnx(请参见PyTorch export onnx)
  • 使用onnxruntime web将生成的onnx与JavaScript一起使用(详见onnxruntime.ai)

如果有人知道更多选项,请添加。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接