/RV-dilatation-DL

The repository of the source code of "Automated Recognition of Right Ventricular Dilatation and Prognosis Correlation"

Primary LanguageJupyter NotebookMIT LicenseMIT

About

This is the source code of "Automated Recognition of Right Ventricular Dilatation and Prognosis Correlation", implemented with Keras and Tensorflow.

Reproduction and model usage

  1. All pretrained classification models were stored under the "model" folder.
  2. run.ipynb: notebook of how you load and use the models or train your own model.
  3. For model architecture and training, please refer to the "src" folder.

PAH prognosis and survival analysis

Application was shown in survival_analysis.ipynb.

Grad-CAM

160019P48701_31_GradCAM

Segmentation

The segmentation results came from our pretrained model. Adding both segmentation channels of left and right ventricle, the input arrays become larger. We trained the A4C-seg model with these reproduced data, which is visualized and shown as the following gif. seg