groupcnn

Group equivariant CNN in pytorch.

This is a partial implementation of neural modules from the paper

'Group Equivariant Convolutional Networks'

by T. Cohen and M. Welling (2016).

So far only rotation equivariant convolutions (P4) are implemented.