ubclaunchpad / physiospace-react

The Physio Space React App
https://thephysio.space/
3 stars 3 forks source link

Look into alternate ML models #24

Open SophieMBerger opened 4 years ago

jackyzha0 commented 4 years ago

Ok after doing a bit more research, there's two main parts we need to implement 1: Stacked Hourglass Network for 2D Pose Prediction (arxiv.org/abs/1603.06937) Key takeaways can be found here: https://github.com/aleju/papers/blob/master/neural-nets/Stacked_Hourglass_Networks_for_Human_Pose_Estimation.md

Train on MPII: http://human-pose.mpi-inf.mpg.de/ Batch norm, single hourglass, RMSProp

2: Simple Linear NN with Batch Norm and Dropout following the work here (https://github.com/una-dinosauria/3d-pose-baseline)

SophieMBerger commented 4 years ago

@jackyzha0 Thanks so much for digging into this! It sounds pretty doable to me, what do you think?

jackyzha0 commented 4 years ago

@SophieMBerger definitely doable :)) I've started implementing the stacked hourglass model in Keras and I'll put up a PR once ive made some progress