Skip to content

WACV2021 - A Weakly Supervised Consistency-based Learning Method for COVID-19 Segmentation in CT Images

License

Notifications You must be signed in to change notification settings

IssamLaradji/covid19_weak_supervision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ccec30a · Aug 1, 2021

History

31 Commits
Jul 31, 2021
Aug 1, 2021
Jun 30, 2020
Jul 31, 2021
Jul 31, 2021
Sep 17, 2020
Jun 30, 2020
Aug 1, 2021
Jul 31, 2021
Jun 30, 2020
Aug 1, 2021

Repository files navigation

A Weakly Supervised Consistency-based Learning Method for COVID-19 Segmentation in CT Images [Paper]

Accepted at WACV2021 Conference. Try the code in this colab.

1. Download Datasets

COVID19

COVID19 V2

COVID19 V3

2. Train & Validate

Run the following command to reproduce the experiments in the paper:

python trainval.py -e weakly_covid19_${DATASET}_${SPLIT} -sb ${SAVEDIR_BASE} -d ${DATADIR} -r 1

The variables (${...}) can be substituted with the following values:

  • DATASET (the COVID dataset): v1, v2, or v3
  • SPLIT (the dataset split): mixed_c2, sep_c2
  • SAVEDIR_BASE: Absolute path to where results will be saved
  • DATADIR: Absolute path containing the downloaded datasets

Experiment hyperparameters are defined in ./exp_configs/weakly_exps.py

3. Visualize the Results

Open results.ipynb for visualization.

Cite

@article{laradji2020weakly,
  title={A Weakly Supervised Consistency-based Learning Method for COVID-19 Segmentation in CT Images},
  author={Laradji, Issam and Rodriguez, Pau and Manas, Oscar and Lensink, Keegan and Law, Marco and Kurzman, Lironne and Parker, William and Vazquez, David and Nowrouzezahrai, Derek},
  journal={arXiv preprint arXiv:2007.02180},
  year={2020}
}

About

WACV2021 - A Weakly Supervised Consistency-based Learning Method for COVID-19 Segmentation in CT Images

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published