Training 3D CNN models

Version 1.0.0 (5.53 KB) by Amir Ebrahimi

This code can be used in conjunction with some 3D CNN models for classification.

276 Downloads

Updated 10 Aug 2021

Use 'MRI_ADvsNC' to train a 3D CNN with your data. This code was originaly used for Alzheimer's disease detection using MRI for two classes. In the code, you can change the number of classes and the type of input data.

Some 3D CNN models can be found here:

3D ResNet-18: https://au.mathworks.com/matlabcentral/fileexchange/82585-pre-trained-3d-resnet-18?s_tid=prof_contriblnk

3D ResNet-50: https://au.mathworks.com/matlabcentral/fileexchange/87427-pre-trained-3d-resnet-50?s_tid=prof_contriblnk

3D ResNet-101: https://au.mathworks.com/matlabcentral/fileexchange/87432-pre-trained-3d-resnet-101?s_tid=prof_contriblnk

3D LeNet-5: https://au.mathworks.com/matlabcentral/fileexchange/87442-pre-trained-3d-lenet-5?s_tid=prof_contriblnk

Please cite the following paper if you find these models useful:

Cite As

Ebrahimi, Amir, et al. “Introducing Transfer Learning to 3D ResNet-18 for Alzheimer’s Disease Detection on MRI Images.” 2020 35th International Conference on Image and Vision Computing New Zealand (IVCNZ), IEEE, 2020, doi:10.1109/ivcnz51579.2020.9290616.

View more styles

MATLAB Release Compatibility

Created with R2021a

Compatible with any release

Platform Compatibility

Windows macOS Linux

Tags Add Tags

Acknowledgements

Inspired by: Alzheimer’s Disease Detection using 3D ResNet-18 on MRI, Deep Learning: Transfer Learning in 10 lines of MATLAB Code, Deep Learning Tutorial Series, Deep Learning in 11 Lines of MATLAB Code

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

classification3DAugmentationPipeline

MRI_ADvsNC.mlx

Version	Published	Release Notes
1.0.0	10 Aug 2021		Download

MLA	Ebrahimi, Amir, et al. “Introducing Transfer Learning to 3D ResNet-18 for Alzheimer’s Disease Detection on MRI Images.” 2020 35th International Conference on Image and Vision Computing New Zealand (IVCNZ), IEEE, 2020, doi:10.1109/ivcnz51579.2020.9290616.
APA	Ebrahimi, A., Luo, S., & Chiong, R. (2020). Introducing Transfer Learning to 3D ResNet-18 for Alzheimer’s Disease Detection on MRI Images. In 2020 35th International Conference on Image and Vision Computing New Zealand (IVCNZ). IEEE. Retrieved from https://doi.org/10.1109%2Fivcnz51579.2020.9290616
BibTeX	@inproceedings{Ebrahimi_2020, doi = {10.1109/ivcnz51579.2020.9290616}, url = {https://doi.org/10.1109%2Fivcnz51579.2020.9290616}, year = 2020, month = {nov}, publisher = {{IEEE}}, author = {Amir Ebrahimi and Suhuai Luo and Raymond Chiong}, title = {Introducing Transfer Learning to 3D {ResNet}-18 for Alzheimer's Disease Detection on {MRI} Images}, booktitle = {2020 35th International Conference on Image and Vision Computing New Zealand ({IVCNZ})} }