Summary: Takes in a numpy file containing preprocessed lung scans (each patient has 20 50x50 scan slices), and uses Matplotlib generates 2D lung scan images in real time while training a convolutional neural network in TensorFlow to predict whether or not the patient has lung cancer.
Created initially for the 2017 Data Science Bowl, and adapted into a demo for our AI club at USC (CAIS++).
- Python 3.5 (preferably via Anaconda)
- Numpy
- Tensorflow
- Matplotlib
- Clone the repo locally
- Naviagate to the repo folder
- Run:
python dsb_train_demo.py
- Sit back and enjoy!
- Sentdex, for his CNN-based data science bowl kernel on Kaggle