/SPDNet

GitHub repository for "A Riemannian Network for SPD Matrix Learning", AAAI 2017.

Primary LanguageMATLAB

SPDNet-master

Zhiwu Huang and Luc Van Gool. A Riemannian Network for SPD Matrix Learning, In Proc. AAAI 2017.

Version 1.0, Copyright(c) November, 2017.

Note that the copyright of the manopt toolbox is reserved by https://www.manopt.org/

Usage

Step1: Place the used AFEW SPD data under the folder "./data/afew/". Note that the used HDM05 and PaSC SPD data are also publicly available.

Step2: Launch spdnet_afew.m for a simple example.

Related Work/Implementation

  1. Thanks to Oleg Smirnov who is Sr. Applied Scientist at Amazon, a TensorFlow ManOpt library is released to reproduce our SPDNet.

  2. A NeurIPS 2019 paper "Riemannian batch normalization for SPD neural networks" develops batch normalization layer upon our SPDNet, with the official PyTorch code being publicly available at the 'Supplemental' tab.

  3. A report "Second-order networks in PyTorch" is released.

  4. Thanks to Alireza Davoudi, there is another Python implementation for SPDNet.

  5. A direct extension of our SPDNet for facial emotion recognition is published by CVPR workshop 2018, with the code being available.

How to Cite

If you find this project helpful, please consider citing us as follows:

@inproceedings{huang2017spdnet,
      title = {A Riemannian Network for SPD Matrix Learning},
      author    = {Huang, Zhiwu and
                   Van Gool, Luc},
      year = {2017},
      booktitle = {Association for the Advancement of Artificial Intelligence (AAAI)}
}