Pytorch code for the Automated Prediction of Lung Cancer with 3D Convolutional Neural Networks. Our cancer predictor obtained a ROC AUC of 0.913 and was ranked 1st place at the ISBI 2018 Lung Nodule Malignancy Prediction challenge.
This code was implemented in Python 2.7.*, using PyTorch, Numpy, pandas, sklearn, scipy, skimage and dicom.
To run the code save the folder of each patient with the dicom files (of the ISBI 2018 Lung challenge) in the folder ./data/ISBI-deid-TRAIN/ and run ./test_ISBI.py
To run the code with a different ling CT scan, save the folder with the dicom files in the folder ./data/ISBI-deid-TRAIN/ and run ./test.py. For scans different from the ISBI 2018 Lung challenge dataset, the program will output the score after the predictor (without the mask post-processing).
Download the trained models from this link. Detector model was trained with the LIDC-IDRI dataset and the predictor with the Kaggle DSB2017 dataset.
If the dataset from the ISBI 2018 Lung Nodule Malignancy Prediction challenge is used, the AUC will be printed using the challenge labels. We obtained an AUC ROC of 0.937 using the training challenge dataset for validation. The test AUC (91.3) was obtained in the challenge server with not-public labels.
In folder ./data/sorted_slices_jpgs/ the program will save images of the axial, sagittal and coronal planes of the 30 detected nodules with highest score of each patient.