Implementation of SPDNet [1], a Riemannian network for SPD matrix learning.
- Python 3.6+
- SciPy
- NumPy
- TensorFlow 2.0+
- TensorFlow RiemOpt
Configure gcloud
to use Python 3:
gcloud config set ml_engine/local_python /usr/bin/python3
Train SPDNet locally on the Acted Facial Expression in Wild [2] dataset:
gcloud ai-platform local train \
--module-name spdnet.task \
--package-path . \
-- \
--data-dir data
--job-dir ckpt
- Huang, Zhiwu, and Luc Van Gool. "A riemannian network for SPD matrix learning." Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence. AAAI Press, 2017.
- Dhall, Abhinav, et al. "Acted facial expressions in the wild database." Australian National University, Canberra, Australia, Technical Report TR-CS-11 2 (2011): 1.