有没有办法下载PoseNet的冻结图(.pb文件)?

时间:2018-11-29 23:14:12

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

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

2 个答案:

答案 0 :(得分:1)

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

答案 1 :(得分:1)

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

我编写了一个多人后处理代码的Python版本,该代码使用矢量化scipy / numpy ops加快了部分速度。我没有对这部分进行详尽的测试,但是针对参考图像对各种测试图像进​​行了多次抽查,并将其用于其他来源,它似乎与原始图像相当接近,并且速度更快:)

Python + TF位于https://github.com/rwightman/posenet-python

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

如果您碰巧正在寻找CoreML端口,那么我将从该项目https://github.com/infocom-tpo/PoseNet-CoreML的重量转换代码开始