search

iwyoo / Autoencoder3D

Convolutional 3D autoencoder
MIT License
14 stars 5 forks source link

Hi do you have written 3D convolution autoencoder? #2

Open markofan opened 7 years ago

markofan commented 7 years ago

Sorry to bother you! I want to ask you that do you know how to write 3D convolution autoencoder?

iwyoo commented 7 years ago

class ConvAutoencoder3D(BasicAutoencoder3D): in https://github.com/iwyoo/Autoencoder3D/blob/master/Autoencoder3D.py is 3d Convolutional autoencoder. However, I do not recommend to use or design 3D convolutional autoencoder with 2 reasons (1. Too much memory usage and 2. Too much parameters to train.). Actually, my implementation also works bad.

markofan commented 7 years ago

Thank you! but do you mind show me how to use it? Because my data is MRI(3d data), I want to use a 3D Stacked autoencoder.

markofan commented 7 years ago

Hi I found another 3D convolution autoencoder in github https://github.com/markofan/3d-convolutional-network/blob/master/convnet_3d.py the class CAE3d, so I don't know if you are interested in that, but you can try that!