GitHub - qingwusunny/c3d_attention_mri

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
model_unit.py		model_unit.py
my_attention.py		my_attention.py
my_c3d.py		my_c3d.py
readme.txt		readme.txt
train_multi_gpu.py		train_multi_gpu.py
write_data.py		write_data.py

Repository files navigation

use attention mechanism to AD classification
the data is MRI images from ADNI
the base network is C3D model


write_data.py 
write the MRI data to tfrecord

model_unit.py
the base unit of my model

my_c3d.py
the c3d model adapt to AD classification

my_attention.py
add attention mechanshim to c3d model

train_multi_gpu.py
train the model