keyforce / hand-keypoint-estimation

手部21点关键点检测

Hand-Keypoint-Estimation

Introduction

手部21点关键点识别

TODO

• ResNet34+Finetune
• SqueezeNet+Finetune
• Hourglass
• Openpose+Design Loss
• 更好的效果展示
• 抗遮挡

Dataset

CMU手部数据集（遮挡比较变态）

Hands from Synthetic Data (6546 + 3243 + 2348 + 2124 = 14261 annotations)
└─hand_labels_synth
    ├─output_viz_synth
    ├─synth1（json文件数据缺失指尖5个关键点）
    ├─synth2
    ├─synth3
    └─synth4

Benchmarking

SqueezeNet+Finetune

Finetune = nn.Sequential(
    Flatten(), 
    nn.ReLU(),
    nn.Dropout(0.5),
    nn.Linear(247808, 256),
    #ReLU不能放BN前会导致BN方差计算错误
    nn.BatchNorm1d(256),
    nn.ReLU(),
    nn.Dropout(0.5),
    nn.Linear(256, 42),
    Reshape(-1,21,2),
    nn.Tanh()
    ) 
Total params: 64,172,906
Total trainable params: 64,172,906
Total non-trainable params: 0
Loss function : MSELoss
Epoch : 200
LR : 0.01->0.0001
Train Loss end : 0.010500	
Valid Loss end : 0.012454

CPU上0.0234s一张图片

GPU-2070Ti上0.00727s一张图片

hand-keypoint-estimation's Issues

项目如何运行的？

你好，感谢你开源代码，文档方面不太详细，请问项目怎么运行的？
看到了很多demo文件，不是很清楚作用是什么？有没有训练好的模型文件？

你好，demo/Openpose.py里面有几个函数显示找不到，请问都引用的哪几个库，版本是什么？

你好，demo/Openpose.py里面以下几个函数显示找不到
Flatten
get_transforms
Tensor
PointsItemList
ResizeMethod
imagenet_stats
Learner
请问是都是哪个库下的？库版本是多少？

模型相关

您好，感谢开源代码，想问一下训练得到的模型有多大，精度怎样