/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 AFEW SPD data under the folder "./data/afew/". The AFEW SPD data can be downloaded from

  1. https://data.vision.ee.ethz.ch/zzhiwu/ManifoldNetData/SPDData/AFEW_SPD_data.zip
  2. https://data.vision.ee.ethz.ch/zzhiwu/ManifoldNetData/SPDData/HDM05_SPDData.zip
  3. https://data.vision.ee.ethz.ch/zzhiwu/ManifoldNetData/SPDData/PaSC_SPDData.zip

Step2: Launch spdnet_afew.m for a simple example.

If you find any bugs, please contact me via zhiwu.huang@vision.ee.ethz.ch

Thanks to Alireza Davoudi, there is a Python implementation for SPDNet at https://github.com/adavoudi/spdnet