Skip to content

3D image classification using CNN (Convolutional Neural Network)

Notifications You must be signed in to change notification settings

jibikbam/CNN-3D-images-Tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 

Repository files navigation

CNN 3D Images using Tensorflow

  • Goal: MRI classification task using CNN (Convolutional Neural Network)

  • Code Dependency: Tensorflow 1.0, Anaconda 4.3.8, Python 2.7

  • Difficulty in learning a model from 3D medical images

    1. Data size is too big. e.g., 218x182x218 or 256x256x40
    2. There is only limited number of data. In other words, training size is too small.
    3. All image looks very similar and only have subtle difference between subjects.
  • Possible solutions

    1. Be equipped with good machine especially the RAM
    2. Downsample images in the preprocessing
    3. Data augmentation e.g., rotate, shift, combination
    4. Transfer learning