有没有办法下载PoseNet的冻结图（.pb文件）？

Question

我打算在python中使用posenet，而不是在browser中，因为我需要将模型作为一个冻结的图来进行推断。有办法吗？在

Answer 1

我们目前没有公开推断的冻结图，但是您可以下载资产并在节点.js环境。在

Answer 2

我在假期里把Google的tfjsposenet移植到Python上。存储库中的演示应用程序会自动下载权重、冻结图形并保存到模型文件中。你可以抓取这个模型并在任何TF变体中使用。在

我编写了一个Python版本的多人后处理代码，它使用矢量化的scipy/numpy操作来加速一些部分。我没有对这一部分做过详尽的测试，但是根据参考资料对各种测试图像进行了多次抽查，并将其用于其他一些来源，这似乎相当接近原始图像，而且速度更快：）

Python+TF athttps://github.com/rwightman/posenet-python

我还在https://github.com/rwightman/posenet-pytorch进行了Pythorch转换

如果您碰巧正在寻找CoreML端口，我从这个项目的权重转换代码开始https://github.com/infocom-tpo/PoseNet-CoreML