我们正在尝试将此处描述的 WASM + SIMD + 多线程用于 bodypix。链接到文章的示例网页显示了我使用 SIMD + 多线程的 MacBook Pro 2018。但是,我们在尝试将它用于 bodypix 的目的时遇到了麻烦。
我们正在使用电子 10.1 (Chromium 85.0.4183.84) 运行它,并尝试使用 node 运行它,并且还作为在 index.html 中加载的网页作为上述文章中描述的脚本标签,但似乎在每个测试中几乎不使用 GPU而且体验比在CPU上跑tensorflow还要差。我们似乎缺少一些东西来利用它 SIMD + 多线程。
在 node 中运行时,这是 yarn package.json 包含的内容:
"@tensorflow-models/body-pix": "^2.0.5",
"@tensorflow-models/mobilenet": "^2.0.4",
"@tensorflow/tfjs-backend-wasm": "^2.3.0",
"@tensorflow/tfjs-core": "^2.3.0",
"@tensorflow/tfjs-node": "^2.3.0",
我们确实验证了后端在运行时设置为 wasm。
任何人都有任何想法可能出了什么问题?