Tensorflow 2.0实时多人姿势估计
什么是新的
2020年10月5日
将模型转换为新的
在Mobilenet V3基础上添加了一个新的openpose模型。
添加了对库依赖
标记为“ v1.0”的旧代码可用。
此存储库包含keras_Realtime_Multi-Person_Pose_Estimation项目的新升级版本,以及一些其他脚本和新模型。
我在Tensorboard中添加了最终热图和paf的可视化。 每100次迭代,会将一张图像传递给模型。 预测的热图和paf记录在Tensorboard中。 您可以每隔几个小时检查一次直观的预测显示,因为它可以很好地了解训练的执行情况。
脚本和笔记本
该项目包含以下脚本和jupyter笔记本:
train_singlenet_mobilenetv3.py-本文提出的用于新模型的训练代码,。 我用Mobilenet V3替换了VGG。 简化模型,只有3 pafs和1热图。
train_2br_vgg.py-旧的CMU模型的训练代码(2017)。 这是旧仓库keras_Realtime_Multi-Person_Pose_E
1