Skip to content

Latest commit

 

History

History
54 lines (25 loc) · 927 Bytes

README.md

File metadata and controls

54 lines (25 loc) · 927 Bytes

PAC18

sMRI based depression classification

Directory Structure

"./models" : for all saved models (in gitignore)

"./raw_data/pac2018_data" : for training data (in gitignore)

"./processed_data/hdf5_files" : for training data after pre processing if required (in gitignore)\

Packages Required

jupyter

pytorch

pandas

simple ITK

nibabel

About ./src/*

  • DataAugment : Contains all required functions for 3D data augmentations

  • DataGenerator : Contains function to load data batch wise..

  • DensenetModel : Contains 3D Densenet model.....

  • Trainer_Tester : Contains train and test class with all train and test strategy

Working....

  • First use Create_Data.ipynb to create prpcessed data

  • Use ./src/main.py to train/test model

Contact

If any comments or issues, pull requests/issues are Welcomed....

Thankyou