pytorch model 的前端运行。
参考资料
使用
首先把项目 clone 下来,然后在建立文件夹 src/assets/ 文件夹。
接下来,我们就要生成 model.onnx ,可以直接里面的 onnx_model.py 文件,会自动下载 pth 模型。根据文档中所说,该模型是 alex 模型。
由于,py 下载可能很慢,所以,可以直接从链接上下载。
下载下来的文件中,放在缓存位置。「该位置在你运行 py 文件的时候会给出」
至于什么是 alex 模型,请参考
下载下来模型之后,运行 py 文件,会自动转成 onnx 模型,然后放在 src/assets/ 中。
然后直接执行
npm run serve就能够在页面上看到效果了。
 
        