/SPDNet

This is an unofficial PyTorch implementation for paper "A Riemannian Network for SPD Matrix Learning", AAAI 2017

Primary LanguagePython

SPD_Net

implement Riemannian gradient in pytorch 0.3.1

The original code is in matlab (https://github.com/zzhiwu/SPDNet)

The network is as follow

image

For more detailed information, please resort to the papaer 'A Riemannian Network for SPD Matrix Learning'.

Usage:

Step1: Place the AFEW SPD data under the folder "./data/afew/". The AFEW SPD data can be downloaded from http://www.vision.ee.ethz.ch/~zzhiwu/

Step2: run spdnet_afew.py to train the model