繁体   English   中英

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

[英]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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM