Skip to content

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

Notifications You must be signed in to change notification settings

YirongMao/SPDNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages