├── .gitignore ├── DATA.md ├── FINETUNE.md ├── PRETRAIN.md ├── README.md ├── data_utils ├── __init__.py ├── chest_dset.py ├── chexpert.py ├── data_path.py ├── jsrt.py ├── mimic.py ├── nih.py ├── randaugment.py ├── rsna.py ├── siim.py ├── vindr.py ├── world_model_transform.py ├── xray_transform.py └── xray_transform_add.py ├── imgs └── fig1.png ├── masks ├── __init__.py ├── blockwise.py ├── data2vec.py ├── default.py ├── extra_multiblock.py ├── multi_multiblock.py ├── multi_multiblock_3d.py ├── multi_multiblock_list.py ├── multi_multiblock_v2.py ├── multiblock.py ├── random.py └── utils.py ├── models ├── __init__.py ├── attentive_pooler.py ├── finetune.py ├── jepa.py ├── jepa_add.py ├── jepa_vit.py ├── jepa_vit_3d.py ├── jepa_vit_add.py ├── models_vit.py ├── pos_embed.py ├── roi.py ├── sscale.py ├── unet_adapter.py ├── unet_adapter_conv.py ├── unigrad.py ├── upernet.py └── utils.py ├── opts.py ├── requirements.txt ├── train_finetune.py ├── train_finetune_seg.py ├── train_jepa.py └── util ├── __init__.py ├── crop.py ├── datasets.py ├── lars.py ├── logger.py ├── lr_decay.py ├── lr_sched.py ├── metrics.py ├── misc.py └── mixup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/.gitignore -------------------------------------------------------------------------------- /DATA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/DATA.md -------------------------------------------------------------------------------- /FINETUNE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/FINETUNE.md -------------------------------------------------------------------------------- /PRETRAIN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/PRETRAIN.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/README.md -------------------------------------------------------------------------------- /data_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/__init__.py -------------------------------------------------------------------------------- /data_utils/chest_dset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/chest_dset.py -------------------------------------------------------------------------------- /data_utils/chexpert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/chexpert.py -------------------------------------------------------------------------------- /data_utils/data_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/data_path.py -------------------------------------------------------------------------------- /data_utils/jsrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/jsrt.py -------------------------------------------------------------------------------- /data_utils/mimic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/mimic.py -------------------------------------------------------------------------------- /data_utils/nih.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/nih.py -------------------------------------------------------------------------------- /data_utils/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/randaugment.py -------------------------------------------------------------------------------- /data_utils/rsna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/rsna.py -------------------------------------------------------------------------------- /data_utils/siim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/siim.py -------------------------------------------------------------------------------- /data_utils/vindr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/vindr.py -------------------------------------------------------------------------------- /data_utils/world_model_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/world_model_transform.py -------------------------------------------------------------------------------- /data_utils/xray_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/xray_transform.py -------------------------------------------------------------------------------- /data_utils/xray_transform_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/data_utils/xray_transform_add.py -------------------------------------------------------------------------------- /imgs/fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/imgs/fig1.png -------------------------------------------------------------------------------- /masks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/__init__.py -------------------------------------------------------------------------------- /masks/blockwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/blockwise.py -------------------------------------------------------------------------------- /masks/data2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/data2vec.py -------------------------------------------------------------------------------- /masks/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/default.py -------------------------------------------------------------------------------- /masks/extra_multiblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/extra_multiblock.py -------------------------------------------------------------------------------- /masks/multi_multiblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/multi_multiblock.py -------------------------------------------------------------------------------- /masks/multi_multiblock_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/multi_multiblock_3d.py -------------------------------------------------------------------------------- /masks/multi_multiblock_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/multi_multiblock_list.py -------------------------------------------------------------------------------- /masks/multi_multiblock_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/multi_multiblock_v2.py -------------------------------------------------------------------------------- /masks/multiblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/multiblock.py -------------------------------------------------------------------------------- /masks/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/random.py -------------------------------------------------------------------------------- /masks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/masks/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/attentive_pooler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/attentive_pooler.py -------------------------------------------------------------------------------- /models/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/finetune.py -------------------------------------------------------------------------------- /models/jepa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/jepa.py -------------------------------------------------------------------------------- /models/jepa_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/jepa_add.py -------------------------------------------------------------------------------- /models/jepa_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/jepa_vit.py -------------------------------------------------------------------------------- /models/jepa_vit_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/jepa_vit_3d.py -------------------------------------------------------------------------------- /models/jepa_vit_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/jepa_vit_add.py -------------------------------------------------------------------------------- /models/models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/models_vit.py -------------------------------------------------------------------------------- /models/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/pos_embed.py -------------------------------------------------------------------------------- /models/roi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/roi.py -------------------------------------------------------------------------------- /models/sscale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/sscale.py -------------------------------------------------------------------------------- /models/unet_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/unet_adapter.py -------------------------------------------------------------------------------- /models/unet_adapter_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/unet_adapter_conv.py -------------------------------------------------------------------------------- /models/unigrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/unigrad.py -------------------------------------------------------------------------------- /models/upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/upernet.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/models/utils.py -------------------------------------------------------------------------------- /opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/opts.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/train_finetune.py -------------------------------------------------------------------------------- /train_finetune_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/train_finetune_seg.py -------------------------------------------------------------------------------- /train_jepa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/train_jepa.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/crop.py -------------------------------------------------------------------------------- /util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/datasets.py -------------------------------------------------------------------------------- /util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/lars.py -------------------------------------------------------------------------------- /util/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/logger.py -------------------------------------------------------------------------------- /util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/lr_decay.py -------------------------------------------------------------------------------- /util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/lr_sched.py -------------------------------------------------------------------------------- /util/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/metrics.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/CheXWorld/HEAD/util/mixup.py --------------------------------------------------------------------------------