Skip to content

Implementation of Quaternion Convolutional Neural Networks

Notifications You must be signed in to change notification settings

skn123/QuaternionCNN_Keras

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

QuaternionCNN_Keras

Implementation of Quaternion Convolutional Neural Networks

Requirements

Keras(The writer's version is 2.1.3, with Tensorflow backend 1.4.1)

Numpy

Scipy

PIL

Scikit-image

quaternion_layers

It contains implementation of quaternion version of convolutional layer and fully-connected layer, called QConv and QDense. They can be applied in same way as normal layers, note that they will consider the inputs and outputs as pure quaternions, which means the number of input/output channels must be multiple of three.

This code is heavily borrowed from Deep Complex Networks

cifar10_cnn.py

It's the one of the examples that Keras gives. The layers are replaced by quaternion layers.

denoising.py

It uses a U-net for denoising. A training set and a validation set are needed to run this file. The input size is 128x128.

The training images should be in a folder named 'dataset' and must be at least 128x128. 128x128 patches will be randomly cropped when training.

The validation images should be in a folder named 'validation_split' and must be 128x128.

Citation

Please cite our work as

@inproceedings{zhu2018quaternion,
  title={Quaternion Convolutional Neural Networks},
  author={Xuanyu Zhu, Yi Xu, Hongteng Xu and Changjian Chen},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  year={2018}
}

About

Implementation of Quaternion Convolutional Neural Networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%