├── .gitignore ├── Data ├── subjectSplit_curated.npy └── training_data_curated_60_openpose │ └── infoData_60_openpose.npy ├── Images └── MarkerEnhancer.png ├── LICENSE ├── README.md ├── data_generator.py ├── dataman.py ├── datasets.py ├── models.py ├── reference_models ├── linear │ ├── arm │ │ ├── mean.npy │ │ ├── metadata.json │ │ ├── model.json │ │ ├── std.npy │ │ └── weights.h5 │ └── body │ │ ├── mean.npy │ │ ├── metadata.json │ │ ├── model.json │ │ ├── std.npy │ │ └── weights.h5 ├── lstm │ ├── arm │ │ ├── mean.npy │ │ ├── metadata.json │ │ ├── model.json │ │ ├── std.npy │ │ └── weights.h5 │ └── body │ │ ├── mean.npy │ │ ├── metadata.json │ │ ├── model.json │ │ ├── std.npy │ │ └── weights.h5 └── transformer │ ├── arm │ ├── mean.npy │ ├── metadata.json │ ├── saved_model.pb │ ├── std.npy │ └── variables │ │ ├── variables.data-00000-of-00001 │ │ └── variables.index │ └── body │ ├── mean.npy │ ├── metadata.json │ ├── saved_model.pb │ ├── std.npy │ └── variables │ ├── variables.data-00000-of-00001 │ └── variables.index ├── requirements.txt ├── settings.py ├── test_data ├── drop_vertical_jump.trc ├── sit_to_stand.trc └── squats.trc ├── test_trained_models.py ├── train_lstm_linear.py ├── train_transformer.py ├── transformer_model.py └── utilities.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/.gitignore -------------------------------------------------------------------------------- /Data/subjectSplit_curated.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/Data/subjectSplit_curated.npy -------------------------------------------------------------------------------- /Data/training_data_curated_60_openpose/infoData_60_openpose.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/Data/training_data_curated_60_openpose/infoData_60_openpose.npy -------------------------------------------------------------------------------- /Images/MarkerEnhancer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/Images/MarkerEnhancer.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/README.md -------------------------------------------------------------------------------- /data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/data_generator.py -------------------------------------------------------------------------------- /dataman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/dataman.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/datasets.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/models.py -------------------------------------------------------------------------------- /reference_models/linear/arm/mean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/arm/mean.npy -------------------------------------------------------------------------------- /reference_models/linear/arm/metadata.json: -------------------------------------------------------------------------------- 1 | {"reference_marker": "midHip"} -------------------------------------------------------------------------------- /reference_models/linear/arm/model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/arm/model.json -------------------------------------------------------------------------------- /reference_models/linear/arm/std.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/arm/std.npy -------------------------------------------------------------------------------- /reference_models/linear/arm/weights.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/arm/weights.h5 -------------------------------------------------------------------------------- /reference_models/linear/body/mean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/body/mean.npy -------------------------------------------------------------------------------- /reference_models/linear/body/metadata.json: -------------------------------------------------------------------------------- 1 | {"reference_marker": "midHip"} -------------------------------------------------------------------------------- /reference_models/linear/body/model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/body/model.json -------------------------------------------------------------------------------- /reference_models/linear/body/std.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/body/std.npy -------------------------------------------------------------------------------- /reference_models/linear/body/weights.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/linear/body/weights.h5 -------------------------------------------------------------------------------- /reference_models/lstm/arm/mean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/arm/mean.npy -------------------------------------------------------------------------------- /reference_models/lstm/arm/metadata.json: -------------------------------------------------------------------------------- 1 | {"reference_marker": "midHip"} -------------------------------------------------------------------------------- /reference_models/lstm/arm/model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/arm/model.json -------------------------------------------------------------------------------- /reference_models/lstm/arm/std.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/arm/std.npy -------------------------------------------------------------------------------- /reference_models/lstm/arm/weights.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/arm/weights.h5 -------------------------------------------------------------------------------- /reference_models/lstm/body/mean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/body/mean.npy -------------------------------------------------------------------------------- /reference_models/lstm/body/metadata.json: -------------------------------------------------------------------------------- 1 | {"reference_marker": "midHip"} -------------------------------------------------------------------------------- /reference_models/lstm/body/model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/body/model.json -------------------------------------------------------------------------------- /reference_models/lstm/body/std.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/body/std.npy -------------------------------------------------------------------------------- /reference_models/lstm/body/weights.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/lstm/body/weights.h5 -------------------------------------------------------------------------------- /reference_models/transformer/arm/mean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/arm/mean.npy -------------------------------------------------------------------------------- /reference_models/transformer/arm/metadata.json: -------------------------------------------------------------------------------- 1 | {"reference_marker": "midHip"} -------------------------------------------------------------------------------- /reference_models/transformer/arm/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/arm/saved_model.pb -------------------------------------------------------------------------------- /reference_models/transformer/arm/std.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/arm/std.npy -------------------------------------------------------------------------------- /reference_models/transformer/arm/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/arm/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /reference_models/transformer/arm/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/arm/variables/variables.index -------------------------------------------------------------------------------- /reference_models/transformer/body/mean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/body/mean.npy -------------------------------------------------------------------------------- /reference_models/transformer/body/metadata.json: -------------------------------------------------------------------------------- 1 | {"reference_marker": "midHip"} -------------------------------------------------------------------------------- /reference_models/transformer/body/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/body/saved_model.pb -------------------------------------------------------------------------------- /reference_models/transformer/body/std.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/body/std.npy -------------------------------------------------------------------------------- /reference_models/transformer/body/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/body/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /reference_models/transformer/body/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/reference_models/transformer/body/variables/variables.index -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/requirements.txt -------------------------------------------------------------------------------- /settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/settings.py -------------------------------------------------------------------------------- /test_data/drop_vertical_jump.trc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/test_data/drop_vertical_jump.trc -------------------------------------------------------------------------------- /test_data/sit_to_stand.trc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/test_data/sit_to_stand.trc -------------------------------------------------------------------------------- /test_data/squats.trc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/test_data/squats.trc -------------------------------------------------------------------------------- /test_trained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/test_trained_models.py -------------------------------------------------------------------------------- /train_lstm_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/train_lstm_linear.py -------------------------------------------------------------------------------- /train_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/train_transformer.py -------------------------------------------------------------------------------- /transformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/transformer_model.py -------------------------------------------------------------------------------- /utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanfordnmbl/marker-enhancer/HEAD/utilities.py --------------------------------------------------------------------------------