├── Fine-tuning └── README.md ├── LICENSE ├── README.md ├── evaluation ├── dataset.py ├── evaluation_funcs.py ├── extract_features.py ├── random_data.py ├── train_cls_decoder.py ├── train_cls_multi_decoder.py ├── train_forecasting_decoder.py ├── train_landmark_decoder.py ├── train_metric_reg_multi_decoder.py ├── train_seg_decoder.py └── transforms.py ├── finetune_visionfm_for_multiclass_classification.py ├── imagen ├── requirements.txt └── src │ ├── sample.py │ └── train.py ├── inference_visionfm_for_multiclass_classification.py ├── loader.py ├── main_pretrain.py ├── models ├── __init__.py ├── head.py ├── unetr_head.py └── vision_transformer.py ├── requirements.txt ├── resource └── visionfm_dataset_agreement_form.pdf ├── train_vitb_fundus.sh └── utils.py /Fine-tuning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/Fine-tuning/README.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/README.md -------------------------------------------------------------------------------- /evaluation/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/dataset.py -------------------------------------------------------------------------------- /evaluation/evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/evaluation_funcs.py -------------------------------------------------------------------------------- /evaluation/extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/extract_features.py -------------------------------------------------------------------------------- /evaluation/random_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/random_data.py -------------------------------------------------------------------------------- /evaluation/train_cls_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/train_cls_decoder.py -------------------------------------------------------------------------------- /evaluation/train_cls_multi_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/train_cls_multi_decoder.py -------------------------------------------------------------------------------- /evaluation/train_forecasting_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/train_forecasting_decoder.py -------------------------------------------------------------------------------- /evaluation/train_landmark_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/train_landmark_decoder.py -------------------------------------------------------------------------------- /evaluation/train_metric_reg_multi_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/train_metric_reg_multi_decoder.py -------------------------------------------------------------------------------- /evaluation/train_seg_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/train_seg_decoder.py -------------------------------------------------------------------------------- /evaluation/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/evaluation/transforms.py -------------------------------------------------------------------------------- /finetune_visionfm_for_multiclass_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/finetune_visionfm_for_multiclass_classification.py -------------------------------------------------------------------------------- /imagen/requirements.txt: -------------------------------------------------------------------------------- 1 | imagen-pytorch 2 | -------------------------------------------------------------------------------- /imagen/src/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/imagen/src/sample.py -------------------------------------------------------------------------------- /imagen/src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/imagen/src/train.py -------------------------------------------------------------------------------- /inference_visionfm_for_multiclass_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/inference_visionfm_for_multiclass_classification.py -------------------------------------------------------------------------------- /loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/loader.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .vision_transformer import * -------------------------------------------------------------------------------- /models/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/models/head.py -------------------------------------------------------------------------------- /models/unetr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/models/unetr_head.py -------------------------------------------------------------------------------- /models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/models/vision_transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/requirements.txt -------------------------------------------------------------------------------- /resource/visionfm_dataset_agreement_form.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/resource/visionfm_dataset_agreement_form.pdf -------------------------------------------------------------------------------- /train_vitb_fundus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/train_vitb_fundus.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ABILab-CUHK/VisionFM/HEAD/utils.py --------------------------------------------------------------------------------