├── .gitignore ├── LICENSE ├── README.md ├── live ├── .gitignore ├── README.md └── receive_viz_and_save.py ├── media ├── IMUPoser_github.png └── IMUVisualizer.png ├── requirements.txt ├── scripts ├── 1. Preprocessing │ ├── 1. preprocess_all.py │ └── 2. preprocess_all_to_imuposer_at_25fps.py ├── 2. Train │ ├── 1. Train Global Model.py │ └── run_combos.sh └── 3. Dataset Walkthrough.ipynb └── src ├── imuposer ├── __init__.py ├── config.py ├── datasets │ ├── __init__.py │ ├── globalModelDataset.py │ ├── globalModelDatasetFineTuneDIP.py │ └── utils.py ├── math │ ├── __init__.py │ ├── angular.py │ ├── general.py │ └── spatial.py ├── models │ ├── LSTMs │ │ ├── IMUPoser_Model.py │ │ ├── IMUPoser_Model_FineTune.py │ │ ├── RNN.py │ │ └── __init__.py │ ├── __init__.py │ ├── loss_functions.py │ └── utils.py ├── smpl │ ├── __init__.py │ ├── basicmodel_m_lbs_10_207_0_v1.0.0.pkl │ ├── parametricModel.py │ └── smpl_vert_segmentation.py └── utils.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/README.md -------------------------------------------------------------------------------- /live/.gitignore: -------------------------------------------------------------------------------- 1 | study_data 2 | -------------------------------------------------------------------------------- /live/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/live/README.md -------------------------------------------------------------------------------- /live/receive_viz_and_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/live/receive_viz_and_save.py -------------------------------------------------------------------------------- /media/IMUPoser_github.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/media/IMUPoser_github.png -------------------------------------------------------------------------------- /media/IMUVisualizer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/media/IMUVisualizer.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/1. Preprocessing/1. preprocess_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/scripts/1. Preprocessing/1. preprocess_all.py -------------------------------------------------------------------------------- /scripts/1. Preprocessing/2. preprocess_all_to_imuposer_at_25fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/scripts/1. Preprocessing/2. preprocess_all_to_imuposer_at_25fps.py -------------------------------------------------------------------------------- /scripts/2. Train/1. Train Global Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/scripts/2. Train/1. Train Global Model.py -------------------------------------------------------------------------------- /scripts/2. Train/run_combos.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/scripts/2. Train/run_combos.sh -------------------------------------------------------------------------------- /scripts/3. Dataset Walkthrough.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/scripts/3. Dataset Walkthrough.ipynb -------------------------------------------------------------------------------- /src/imuposer/__init__.py: -------------------------------------------------------------------------------- 1 | from .smpl import * 2 | -------------------------------------------------------------------------------- /src/imuposer/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/config.py -------------------------------------------------------------------------------- /src/imuposer/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/datasets/__init__.py -------------------------------------------------------------------------------- /src/imuposer/datasets/globalModelDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/datasets/globalModelDataset.py -------------------------------------------------------------------------------- /src/imuposer/datasets/globalModelDatasetFineTuneDIP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/datasets/globalModelDatasetFineTuneDIP.py -------------------------------------------------------------------------------- /src/imuposer/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/datasets/utils.py -------------------------------------------------------------------------------- /src/imuposer/math/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/math/__init__.py -------------------------------------------------------------------------------- /src/imuposer/math/angular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/math/angular.py -------------------------------------------------------------------------------- /src/imuposer/math/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/math/general.py -------------------------------------------------------------------------------- /src/imuposer/math/spatial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/math/spatial.py -------------------------------------------------------------------------------- /src/imuposer/models/LSTMs/IMUPoser_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/models/LSTMs/IMUPoser_Model.py -------------------------------------------------------------------------------- /src/imuposer/models/LSTMs/IMUPoser_Model_FineTune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/models/LSTMs/IMUPoser_Model_FineTune.py -------------------------------------------------------------------------------- /src/imuposer/models/LSTMs/RNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/models/LSTMs/RNN.py -------------------------------------------------------------------------------- /src/imuposer/models/LSTMs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/models/LSTMs/__init__.py -------------------------------------------------------------------------------- /src/imuposer/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .LSTMs import * 2 | -------------------------------------------------------------------------------- /src/imuposer/models/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/models/loss_functions.py -------------------------------------------------------------------------------- /src/imuposer/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/models/utils.py -------------------------------------------------------------------------------- /src/imuposer/smpl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/smpl/__init__.py -------------------------------------------------------------------------------- /src/imuposer/smpl/basicmodel_m_lbs_10_207_0_v1.0.0.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/smpl/basicmodel_m_lbs_10_207_0_v1.0.0.pkl -------------------------------------------------------------------------------- /src/imuposer/smpl/parametricModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/smpl/parametricModel.py -------------------------------------------------------------------------------- /src/imuposer/smpl/smpl_vert_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/smpl/smpl_vert_segmentation.py -------------------------------------------------------------------------------- /src/imuposer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/imuposer/utils.py -------------------------------------------------------------------------------- /src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FIGLAB/IMUPoser/HEAD/src/setup.py --------------------------------------------------------------------------------