├── README.md ├── create_features.sh ├── create_patches.sh ├── create_patches_fp.py ├── dataset_csv └── camelyon16.csv ├── datasets ├── dataset_generic.py ├── dataset_h5.py ├── wsi_dali_h5.py └── wsi_dataset.py ├── e2e_train.sh ├── eval.py ├── extract_features_fp.py ├── extract_topK_ROIs.py ├── extract_topk_rois.sh ├── models ├── ABMIL.py ├── DSMIL.py ├── KNN_classifier.py ├── MaxPooling.py ├── MeanPooling.py ├── TransMIL.py ├── model_clam.py ├── model_mil.py ├── resnet_custom.py └── vib_clam_wsi.py ├── poster-cvpr.pdf ├── requirements.txt ├── splits └── task_camelyon16 │ ├── splits_0.csv │ ├── splits_1.csv │ ├── splits_2.csv │ ├── splits_3.csv │ └── splits_4.csv ├── test.sh ├── train.sh ├── train_end2end.py ├── train_stage1.py ├── train_vib.py ├── utils ├── core_utils.py ├── core_utils_end2end.py ├── eval_utils.py ├── fig3.pdf ├── file_utils.py └── utils.py ├── vib_train.sh └── wsi_core ├── WholeSlideImage.py ├── batch_process_utils.py ├── util_classes.py └── wsi_utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/README.md -------------------------------------------------------------------------------- /create_features.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/create_features.sh -------------------------------------------------------------------------------- /create_patches.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/create_patches.sh -------------------------------------------------------------------------------- /create_patches_fp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/create_patches_fp.py -------------------------------------------------------------------------------- /dataset_csv/camelyon16.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/dataset_csv/camelyon16.csv -------------------------------------------------------------------------------- /datasets/dataset_generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/datasets/dataset_generic.py -------------------------------------------------------------------------------- /datasets/dataset_h5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/datasets/dataset_h5.py -------------------------------------------------------------------------------- /datasets/wsi_dali_h5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/datasets/wsi_dali_h5.py -------------------------------------------------------------------------------- /datasets/wsi_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/datasets/wsi_dataset.py -------------------------------------------------------------------------------- /e2e_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/e2e_train.sh -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/eval.py -------------------------------------------------------------------------------- /extract_features_fp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/extract_features_fp.py -------------------------------------------------------------------------------- /extract_topK_ROIs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/extract_topK_ROIs.py -------------------------------------------------------------------------------- /extract_topk_rois.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/extract_topk_rois.sh -------------------------------------------------------------------------------- /models/ABMIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/ABMIL.py -------------------------------------------------------------------------------- /models/DSMIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/DSMIL.py -------------------------------------------------------------------------------- /models/KNN_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/KNN_classifier.py -------------------------------------------------------------------------------- /models/MaxPooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/MaxPooling.py -------------------------------------------------------------------------------- /models/MeanPooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/MeanPooling.py -------------------------------------------------------------------------------- /models/TransMIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/TransMIL.py -------------------------------------------------------------------------------- /models/model_clam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/model_clam.py -------------------------------------------------------------------------------- /models/model_mil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/model_mil.py -------------------------------------------------------------------------------- /models/resnet_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/resnet_custom.py -------------------------------------------------------------------------------- /models/vib_clam_wsi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/models/vib_clam_wsi.py -------------------------------------------------------------------------------- /poster-cvpr.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/poster-cvpr.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/requirements.txt -------------------------------------------------------------------------------- /splits/task_camelyon16/splits_0.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/splits/task_camelyon16/splits_0.csv -------------------------------------------------------------------------------- /splits/task_camelyon16/splits_1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/splits/task_camelyon16/splits_1.csv -------------------------------------------------------------------------------- /splits/task_camelyon16/splits_2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/splits/task_camelyon16/splits_2.csv -------------------------------------------------------------------------------- /splits/task_camelyon16/splits_3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/splits/task_camelyon16/splits_3.csv -------------------------------------------------------------------------------- /splits/task_camelyon16/splits_4.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/splits/task_camelyon16/splits_4.csv -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/test.sh -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 python train_stage1.py --exp_code 'clam_cam16_res50_ostu_init' -------------------------------------------------------------------------------- /train_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/train_end2end.py -------------------------------------------------------------------------------- /train_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/train_stage1.py -------------------------------------------------------------------------------- /train_vib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/train_vib.py -------------------------------------------------------------------------------- /utils/core_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/utils/core_utils.py -------------------------------------------------------------------------------- /utils/core_utils_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/utils/core_utils_end2end.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/fig3.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/utils/fig3.pdf -------------------------------------------------------------------------------- /utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/utils/file_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/utils/utils.py -------------------------------------------------------------------------------- /vib_train.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=7 python train_stage1.py --exp_code 'clam_cam16_res50_vib_mse_infoloss_noavg' --vib -------------------------------------------------------------------------------- /wsi_core/WholeSlideImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/wsi_core/WholeSlideImage.py -------------------------------------------------------------------------------- /wsi_core/batch_process_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/wsi_core/batch_process_utils.py -------------------------------------------------------------------------------- /wsi_core/util_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/wsi_core/util_classes.py -------------------------------------------------------------------------------- /wsi_core/wsi_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/invoker-LL/WSI-finetuning/HEAD/wsi_core/wsi_utils.py --------------------------------------------------------------------------------