Skip to content
/ SPDNet Public
forked from YirongMao/SPDNet

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

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

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%