An end-to-end deep learning framework for predicting hematoma expansion in hemorrhagic stroke patients from CT images
In this respository, you can find the code for the paper "An end-to-end deep learning framework for predicting hematoma expansion in hemorrhagic stroke patients from CT images".
The following python files are available:
-
PatchDataModule_wMask_crossval.py
- This file contains the class handling the dataset - loading the data, preparing patches for train/val/test splits
-
model_wMask.py
- This file contains the code for the model to predict hematoma expansion pattern
-
mask_dvf_loop_crossval.py
- This file contains the main code to run training