【问题标题】:is there a way to download the frozen graph ( .pb file ) of PoseNet?有没有办法下载 PoseNet 的冻结图(.pb 文件)?
【发布时间】:2018-11-29 23:14:11
【问题描述】:

我打算在 python 中而不是在浏览器中使用posenet,因为我需要将模型作为冻结图来进行推理。有没有办法做到这一点?

【问题讨论】:

    标签: python tensorflow neural-network deep-learning tensorflow.js


    【解决方案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

    【讨论】:

      【解决方案2】:

      我们目前没有公开的用于推理的冻结图,但是您可以下载资产并在 Node.js 环境中运行它们。

      【讨论】:

        猜你喜欢
        • 2016-06-15
        • 2020-08-16
        • 2023-02-07
        • 2019-01-04
        • 1970-01-01
        • 1970-01-01
        • 1970-01-01
        • 1970-01-01
        • 2018-12-17
        相关资源
        最近更新 更多