python - 有没有办法下载 PoseNet 的卡住图(.pb 文件)？

Question

我打算在Python中使用posenet，而不是在浏览器中，因为我需要模型作为卡住图来进行推理。有办法做到这一点吗？

Answer 1

我在假期期间将 Google 的 tfjs PoseNet 移植到了 Python。存储库中的演示应用程序会自动下载权重、卡住图形并保存到模型文件。您可以获取此模型并在任何 TF 变体中使用。

我编写了多人后处理代码的 Python 版本，该代码使用矢量化 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 的权重转换代码开始。