DeformCNNAE

2D and 3D deformable CNN Autoencoders on EEG spectrums for classification task ( Major Depressive Disorder vs Helathy Control)

DeformConv3D implementation in deform_conv.py module is used from the repository https://github.com/kondratevakate/3d-deformable-convolutions
DeformConv2D implementation in deform_conv.py module is used from the repository https://github.com/developer0hye/PyTorch-Deformable-Convolution-v2

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
README.md		README.md
cross_val_autoencoder.ipynb		cross_val_autoencoder.ipynb
datasets.py		datasets.py
deform_conv.py		deform_conv.py
early_stopping.py		early_stopping.py
main.py		main.py
models.py		models.py
trainer.py		trainer.py

Provide feedback