You can create the environment via:
pip install -r requirement.txtThe PDGait dataset is download from A public data set of walking. The the data is preprocessed through:
cd lib/data/pdgait
python convert_pd.py
python preprocess_pd.pyThe 3DGait dataset is download from Video-based-gait-analysis. The the data is preprocessed through:
cd lib/data/3dgait
python convert_pd.py
python preprocess_pd.pyThere are three traing stages.
- stage 1: pretrain 3DmotionBERT:
bash pretrain.sh- stage 2: train Gait Denoiser:
bash pretrain_diffusion.sh- stage 3: train Classifier:
pythoon pd_score.pyIt will be released soon.
