hand pose estimation

siwan-tc · Nov 13, 2018 · bea35b7 · bea35b7
1 parent fcf0fe6
commit bea35b7
Show file tree

Hide file tree

Showing 8 changed files with 298 additions and 213 deletions.
diff --git a/README.md b/README.md
@@ -2,16 +2,33 @@
 
 pytorch implementation of [openpose](https://github.com/CMU-Perceptual-Computing-Lab/openpose) including **Body and Hand Pose Estimation**, and the pytorch model is directed converted from [openpose](https://github.com/CMU-Perceptual-Computing-Lab/openpose) caffemodel by [caffemodel2pytorch](https://github.com/vadimkantorov/caffemodel2pytorch). You could implement face keypoint detection in the same way if you are interested in.
 
+### Model Download
+* [dropbox](https://www.dropbox.com/sh/7xbup2qsn7vvjxo/AABWFksdlgOMXR_r5v3RwKRYa?dl=0)
+
+`*.pth` files are pytorch model, you could also download caffemodel file if you want to use caffe as backend.
+
 ### Todo list
 - [x] convert caffemodel to pytorch.
 - [x] Body Pose Estimation.
-- [ ] Hand Pose Estimation.
+- [x] Hand Pose Estimation.
 - [ ] Performance test.
 - [ ] Speed up.
 
 ### Demo
 #### Skeleton
-![](images/skeleton.jpg)
+
+<div align='center'>
+<img src='images/skeleton.jpg'>
+<div/>
 
 #### Body Pose Estimation
-![](images/body_preview.jpg)
+
+<div align='center'>
+<img src='images/body_preview.jpg'>
+<div/>
+
+### Hand Pose Estimation
+
+<div align='center'>
+<img src='images/hand_preview.png'>
+<div/>
diff --git a/images/hand.jpg b/images/hand.jpg
diff --git a/images/hand_preview.png b/images/hand_preview.png
diff --git a/images/hand_preview_estimation.png b/images/hand_preview_estimation.png
diff --git a/notebooks/hand.ipynb b/notebooks/hand.ipynb
diff --git a/python/test.py b/python/test.py
diff --git a/python/test1.py b/python/test1.py
diff --git a/python/util.py b/python/util.py
@@ -33,6 +33,7 @@ def transfer(model, model_weights):
         transfered_model_weights[weights_name] = model_weights['.'.join(weights_name.split('.')[1:])]
     return transfered_model_weights
 
+# draw the body keypoint and lims
 def draw_bodypose(canvas, candidate, subset):
     stickwidth = 4
     limbSeq = [[2, 3], [2, 6], [3, 4], [4, 5], [6, 7], [7, 8], [2, 9], [9, 10], \