有没有办法下载 PoseNet 的冻结图(.pb 文件)?
is there a way to download the frozen graph ( .pb file ) of PoseNet?
我打算在 python 而不是在浏览器中使用 posenet,因为我需要将模型作为冻结图来进行推理。有办法吗?
我们目前没有用于公开推理的冻结图,但是您可以在 Node.js 环境中下载资产和 运行 它们。
我在假期将 Google 的 tfjs PoseNet 移植到了 Python。存储库中的演示应用程序会自动下载权重、冻结图形并保存到模型文件中。您可以获取此模型并在任何 TF 变体中使用。
我写了一个 Python 版本的多人 post 处理代码,它使用矢量化 scipy/numpy 操作来加速一些部分。我没有对这部分进行详尽的测试,但是根据参考对各种测试图像进行了多次抽查,并将其用于其他一些来源,它似乎相当接近原始,而且速度更快:)
Python + TF 在 https://github.com/rwightman/posenet-python
我还在 https://github.com/rwightman/posenet-pytorch
进行了 PyTorch 转换
如果您恰好在某个时候寻找 CoreML 端口,我从这个项目的权重转换代码开始 https://github.com/infocom-tpo/PoseNet-CoreML
我打算在 python 而不是在浏览器中使用 posenet,因为我需要将模型作为冻结图来进行推理。有办法吗?
我们目前没有用于公开推理的冻结图,但是您可以在 Node.js 环境中下载资产和 运行 它们。
我在假期将 Google 的 tfjs PoseNet 移植到了 Python。存储库中的演示应用程序会自动下载权重、冻结图形并保存到模型文件中。您可以获取此模型并在任何 TF 变体中使用。
我写了一个 Python 版本的多人 post 处理代码,它使用矢量化 scipy/numpy 操作来加速一些部分。我没有对这部分进行详尽的测试,但是根据参考对各种测试图像进行了多次抽查,并将其用于其他一些来源,它似乎相当接近原始,而且速度更快:)
Python + TF 在 https://github.com/rwightman/posenet-python
我还在 https://github.com/rwightman/posenet-pytorch
进行了 PyTorch 转换如果您恰好在某个时候寻找 CoreML 端口,我从这个项目的权重转换代码开始 https://github.com/infocom-tpo/PoseNet-CoreML