├── .gitignore ├── README.md ├── configs ├── eval.yaml └── train.yaml ├── data └── J_regressor_h36m.npy ├── eval.py ├── learned_gradient_descent ├── __init__.py ├── data │ ├── AMASS_dataset.py │ ├── PW_dataset.py │ └── __init__.py ├── models │ ├── __init__.py │ └── baseline.py └── utils │ ├── __init__.py │ ├── camera.py │ ├── loss.py │ ├── network.py │ ├── pose_util.py │ ├── smpl.py │ └── transforms.py ├── outputs └── AMASS │ └── Baseline │ └── ckpts │ └── pretrained.ckpt ├── requirements.txt ├── scripts ├── preprocess_3DPW.py └── preprocess_AMASS.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | /data 3 | outputs/ 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/README.md -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /data/J_regressor_h36m.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/data/J_regressor_h36m.npy -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/eval.py -------------------------------------------------------------------------------- /learned_gradient_descent/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /learned_gradient_descent/data/AMASS_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/data/AMASS_dataset.py -------------------------------------------------------------------------------- /learned_gradient_descent/data/PW_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/data/PW_dataset.py -------------------------------------------------------------------------------- /learned_gradient_descent/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/data/__init__.py -------------------------------------------------------------------------------- /learned_gradient_descent/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /learned_gradient_descent/models/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/models/baseline.py -------------------------------------------------------------------------------- /learned_gradient_descent/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /learned_gradient_descent/utils/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/utils/camera.py -------------------------------------------------------------------------------- /learned_gradient_descent/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/utils/loss.py -------------------------------------------------------------------------------- /learned_gradient_descent/utils/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/utils/network.py -------------------------------------------------------------------------------- /learned_gradient_descent/utils/pose_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/utils/pose_util.py -------------------------------------------------------------------------------- /learned_gradient_descent/utils/smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/utils/smpl.py -------------------------------------------------------------------------------- /learned_gradient_descent/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/learned_gradient_descent/utils/transforms.py -------------------------------------------------------------------------------- /outputs/AMASS/Baseline/ckpts/pretrained.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/outputs/AMASS/Baseline/ckpts/pretrained.ckpt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/preprocess_3DPW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/scripts/preprocess_3DPW.py -------------------------------------------------------------------------------- /scripts/preprocess_AMASS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/scripts/preprocess_AMASS.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InpatientJam/Learned-Gradient-Descent/HEAD/train.py --------------------------------------------------------------------------------