- training the backbone
python train.py --dataset CXR14 --data_dir <PATH TO CXR14 DATASET>
- our method
python train.py --experiment TEDM --data_dir <PATH TO JSRT DATASET> --n_labelled_images <TRAINING SET SIZE>
- LEDM method
python train.py --experiment LEDM --data_dir <PATH TO JSRT DATASET> --n_labelled_images <TRAINING SET SIZE>
- LEDMe method
python train.py --experiment LEDMe --data_dir <PATH TO JSRT DATASET> --n_labelled_images <TRAINING SET SIZE>
- baseline method
python train.py --experiment JSRT_baseline --data_dir <PATH TO JSRT DATASET> --n_labelled_images <TRAINING SET SIZE>
-
update
DATADIR
in pathsdataloaders/JSRT.py
,dataloaders/NIH.py
anddataloaders/Montgomery.py
NIHPATH
,NIHFILE
,MONPATH
andMONFILE
in pathsauxiliary/postprocessing/run_tests.py
andauxiliary/postprocessing/testing_shared_weights.py
-
for baseline and LEDM methods, run
python auxiliary/postprocessing/run_tests.py --experiment <PATH TO LOG FOLDER>
- for our method, run
python auxiliary/postprocessing/testing_shared_weights.py --experiment <PATH TO LOG FOLDER>
VS Code notebooks can be found in auxiliary/notebooks_and_reporting
.