Skip to content

This repository deals with analyzing various Neural Network approaches and finding the one with the most accurate reconstruction of motion captured trajectories recorded with missing markers in softwares like Vicon Nexus

chenwuperth/Reconstruction-of-Trajectory-recorded-with-Missing-Markers

 
 

Repository files navigation

Reconstruction of Trajectory with Missing Markers

The project deals with implementing various Neural Network approaches to perform the reconstruction of missing markers in the human motion capture data and as a result finding the one which gives the most accurate recreation.

Prerequisites

  • Python 3
  • Tensorflow >= 1.0
  • keras-tcn
  • numpy
  • matplotlib
  • torchdiffeq
  • torch >= 1.0

Human Motion Capture Data

The pre-processed human motion capture data as well as the base code were extracted from A Neural Network Approach to Missing Marker Reconstruction. In order to use your own Motion Capture data, follow the steps in the Data Preparation section of A Neural Network Approach to Missing Marker Reconstruction repository.

Once the data is available make sure that you put it in data folder ad change the path in code/utils/flags.py appropriately for every implementation.

Implementations

Temporal Convolutional Neural Network

See this Jupyter Notebook or Open In Colab

Temporal Convolutional Neural Network + Wasserstein GAN

See this Jupyter Notebook or Open In Colab

Temporal Convolutional Neural Network + Wasserstein GAN with Gradient Penalty

See this Jupyter Notebook or Open In Colab

Open In Colab

Temporal Convolutional Neural Network + LSTM

See this Jupyter Notebook or Open In Colab

Wasserstein GAN + LSTM

See this Jupyter Notebook or Open In Colab

Note that WGAN + LSTM implementation was ran for 50 epochs instead of the default 100 epochs(for the rest of the networks) as WGAN was not showing any further improvement in test RMSE.

See this Jupyter Notebook or Open In Colab

Conclusions

Temporal Convolutional Neural Network

The training session for TCN started with a very low validation RMSE error (~40) compared to the other networks, however in the subsequent epochs the RMSE error did not improve but stayed the same. At the end of the training session, the test RMSE error was ~32 which was just a slight decrease from the starting validation error. According to our observations, the limitations in the receptive field of the Temporal Convolutional Neural Network is the reason for the above training performance.

Temporal Convolutional Neural Network + Wasserstein GAN

To leverage the WGAN features with TCN which had a hard time training, an ensemble network was trained showing a starting validation RMSE error of ~ 117 and eventually a test RMSE error of ~24 after the training ended. As a result TCN seems to have increased WGAN's test RMSE error. Similar results were observed for TCN + WGAN with Gradient Penalty.

Long Short Term Network

LSTM was trained with the same hyperparameters as mentioned in A Neural Network Approach to Missing Marker Reconstruction. The training session for LSTM started with exceptionally low validation RMSE error of ~14 (test RMSE error for other networks) and ended with test RMSE error of ~2. As expected LSTM is unmatched in time-series sequence to sequence generation task. However, in order to lower down the test RMSE error further, various other ensemble neural networks were devised with a combination of LSTM and another neural network.

Temporal Convolutional Neural Network + Long Short Term Network

Similar to LSTM's cases, the training session here started with a validation RMSE error of ~14, however due to the non-training influence of TCN, the training ended with approximately equal test RMSE error of ~13.

Wasserstein GAN + Long Short Term Network

WGAN+LSTM's training started with a very high validation RMSE error (~117) as expected, nevertheless in the next five training epochs WGAN+LSTM trained very quickly, thus bring down validation RMSE error to ~19. As WGAN+LSTM was not showing much improvements after epoch 5 all the way up to epoch 50, the training session was stopped at epoch 50. This suggests that the current WGAN+LSTM configuration had trained completely and could not go beyond ~15 test RMSE error.

Latent ODEs for Irregularly-Sampled Time Series

  • Latent ODE with RNN encoder:

    • Initial validation MSE error: 0.6534
    • Final validation MSE error: 0.0582
  • ODE-RNN:

    • Initial validation MSE error: 0.4381
    • Final validation MSE error: 0.0567
  • RNN-VAE:

    • Initial validation MSE error: 0.56
    • Final validation MSE error: 0.4709
  • Latent ODE with ODE-RNN encoder and poisson likelihood:

    • Initial validation MSE error: 0.4918
    • Final validation MSE error: 0.0626
  • Latent ODE with ODE-RNN encoder:

    • Initial validation MSE error: 0.4779
    • Final validation MSE error: 0.0616

For more information regarding the above models refer Latent ODEs for Irregularly-Sampled Time Series.

Localization Data for Person Activity Data Set is used in the above case which is different from the previous implementations.

Conclusively, through this project we first experimented with how efficient Generative Adversarial Networks and Temporal Convolutional Neural Networks can be pertaining to sequence-sequence generation task.

Thereafter having found the state of the art Long Short Term Memory Networks, we tried to augment the same however it failed with WGAN and TCN.

Ultimately, we did find a very accurate network: Latent ODEs, as evident from the above validation MSE errors of Latent ODEs in comparison to RNN-VAE or LSTM. Hence Latent ODEs possess a strong potential towards automatic reconstruction of motion capture trajectories recorded with missing markers in softwares like Vicon Nexus.

Note that Latent ODE models are Ordinary Differential Equations-based models differentiating them from their Recurrent Neural Network-based counterparts.

Customization

You can experiment with the hyperparameters of the network by changing the same in the file code/utils/flags.py present in each of the network's folder.

Acknowledgements

  • Kucherenko, Taras, Jonas Beskow, and Hedvig Kjellström. "A neural network approach to missing marker reconstruction in human motion capture." arXiv preprint arXiv:1803.02665 (2018).

  • Rubanova, Yulia, Ricky TQ Chen, and David Duvenaud. "Latent odes for irregularly-sampled time series." arXiv preprint arXiv:1907.03907 (2019).

About

This repository deals with analyzing various Neural Network approaches and finding the one with the most accurate reconstruction of motion captured trajectories recorded with missing markers in softwares like Vicon Nexus

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 53.6%
  • Jupyter Notebook 46.4%