alexkarargyris / Color-Space-Transformation-Network

This is a color space network module that can plug into a neural network.
MIT License
6 stars 2 forks source link

Color Space Transformation Network

This is a color space network module that can plug into a neural network. It is a network layer that adjusts its configuration after training on a colored image dataset. Its output is a 3x3 transformation that is applied to the original 3-channeled images to help increase network's overall classification accuracy.

More technical information about the architecture can be found in the following short paper: http://arxiv.org/abs/1511.01064

It uses Lasagne. Please follow instructions here: http://lasagne.readthedocs.org/en/latest/user/installation.html