Skip to content

Medabid1/ComplexValuedCNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ComplexValuedCNN

Complex CNN in Pytorch, implementation of the paper Deep Complex Networks : https://arxiv.org/abs/1705.09792

Modules implemeted :

  • Complex Conv2d conv
  • Complex ConvTranspose2d
  • Complex AvgPool2d
  • Complex Linear

Activation function :

  • zRelu
  • CRelu

Usage :

  1. Define your Complex input using the class CPLX. x = CPLX(real_part, imaginary_part)
  2. Build your model using src modules.
  3. Use the loss function in utils.py to train your model. if the labels are real valued, set use_magnitude = True

ToDo

  • Complex UNet

About

Complex Valued CNN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages