![](/img/trans.png)
[英]Tensorflow: Load a .pb file and then save it as a frozen graph issues
[英]is there a way to download the frozen graph ( .pb file ) of PoseNet?
我打算在python中而不是在浏览器中使用posenet,因为我需要模型作为冻结图来进行推断。 有没有办法做到这一点?
我们目前没有公开的推理冻结图,但是您可以下载资产并在Node.js环境中运行它们。
在假期期间,我将Google的tfjs PoseNet移植到了Python。 存储库中的演示应用程序会自动下载权重,冻结图形并保存到模型文件。 您可以获取此模型并在任何TF变体中使用。
我编写了多人后处理代码的Python版本,该代码使用矢量化的scipy / numpy ops加快了一些过程。 我没有对这部分进行详尽的测试,但是对各种测试图像进行了大量的抽查,以供参考,并将其用于其他来源,这似乎与原始图像相当接近,并且速度更快:)
https://github.com/rwightman/posenet-python上的 Python + TF
我还在https://github.com/rwightman/posenet-pytorch进行了PyTorch转换
而且,如果您恰巧在某个时候正在寻找CoreML端口,那么我将从该项目的重量转换代码开始https://github.com/infocom-tpo/PoseNet-CoreML
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.