posenet python:Google TensorFlow.js PoseNet的Python端口(实时人体姿势估计) 源码

PoseNet Python 该存储库包含Google TensorFlow.js Posenet模型的纯Python实现(仅限多姿势)。 对于此之后的(略快)PyTorch实现,请参阅( ) 我首先或多或少地逐字修改了JS代码,发现性能很低,因此对一些关键函数(命名为_fast )做了一些矢量化的numpy / scipy版本。 进一步优化是可能的 基本的MobileNet型号在GTX 1080 Ti(或更高版本)上具有200-300 fps的吞吐量 多位置后处理代码使该速率大大降低。 使用快速CPU和GTX 1080+: JS后处理代码的字面翻译将性能降低到大约30fps 我的