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
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