-
Pytorch版MobilePose:一款轻量级单人姿态估计工具
资源介绍
移动姿势
MobilePose是单人2D姿势估计框架的Tiny PyTorch实现。 目的是为训练/推论/评估界面以及数据加载器提供各种数据增强选项。 最终的训练模型可以满足移动设备的基本要求(速度,尺寸,精度)。
用于网络和显示的一些代码来自:
消息!
2021年4月:提供了!
2019年3月:支持以不错的FPS在MacBook上运行!
2019年2月:提供所有预训练的模型文件!
要求
Python 3.7
PyTorch 1.0
评价结果
型号(+ DUC + DSNTNN)
帕玛斯(M)
拖鞋(G)
AP@0.5:0.95
AP@0.5
AR@0.5:0.95
AR@0.5
关联
ResNet18
12.26
1.64
68.2
93.9
79.7
96.7
MobileNetV2
3.91
0.49
67.5
94.9
79.4