├── .gitignore ├── README.md ├── classification ├── README.md ├── datasets │ ├── data_splits │ │ ├── covidx3 │ │ │ ├── test_COVIDx9A.txt │ │ │ └── train_COVIDx9A.txt │ │ └── cxr14 │ │ │ ├── test_official.txt │ │ │ ├── train_official.txt │ │ │ └── val_official.txt │ └── prepare_dataset.py ├── engines │ ├── __init__.py │ └── engine_finetune.py ├── figs │ ├── cls-result.png │ └── x-ray-logo.png ├── models │ ├── models_eva.py │ ├── models_vit.py │ └── rope.py ├── tools │ ├── interpolate14to16.py │ └── preprocess_r50.py ├── train.py ├── train.sh ├── train_files │ ├── dn121 │ │ └── cxr14 │ │ │ ├── dn121_medical_mae_cxr14.sh │ │ │ └── dn121_mocov2_mae_cxr14.sh │ ├── eva_x │ │ ├── chexpert │ │ │ ├── vit_s.sh │ │ │ └── vit_ti.sh │ │ ├── covidx3 │ │ │ └── vit_s.sh │ │ ├── covidx4 │ │ │ └── vit_s.sh │ │ └── cxr14 │ │ │ ├── vit_b.sh │ │ │ ├── vit_s.sh │ │ │ └── vit_ti.sh │ ├── resnet │ │ └── cxr14 │ │ │ ├── r50_biovil.sh │ │ │ ├── r50_medklip.sh │ │ │ └── r50_mgca.sh │ └── vit │ │ └── cxr14 │ │ └── vit_s_medical_mae_cxr14.sh └── utils │ ├── __init__.py │ ├── augment.py │ ├── calculate_cam.py │ ├── custom_transforms.py │ ├── dataloader_med.py │ ├── datasets.py │ ├── eva_utils.py │ ├── lars.py │ ├── lr_decay.py │ ├── lr_sched.py │ ├── misc.py │ ├── multi_label_loss.py │ ├── pos_embed.py │ └── sampler.py ├── eva_x.py ├── figs ├── first_fig_00.png ├── logo.png ├── performance.png ├── pipeline.png ├── poster.png └── radar_chart_00.png ├── grad_cam ├── README.md ├── figs │ └── cam.png ├── get_cams.py ├── get_cams.sh ├── model_cnns.py ├── model_eva.py ├── models_vit.py └── utils.py ├── requirements.txt └── segmentation ├── README.md ├── config ├── __base__ │ ├── datasets │ │ ├── lung.py │ │ ├── rsna.py │ │ ├── shenzhen.py │ │ └── siim.py │ ├── models │ │ ├── fcn_resunet.py │ │ ├── upernet_eva_x_small.py │ │ └── upernet_medical_mae_small.py │ ├── runtime │ │ └── default_runtime.py │ └── schedules │ │ ├── schedule_10ep_adamw.py │ │ ├── schedule_10ep_adamw_conv.py │ │ ├── schedule_50ep_adamw.py │ │ └── schedule_50ep_adamw_conv.py ├── eva_x │ ├── upernet_eva_x_small_lung_10ep.py │ ├── upernet_eva_x_small_rsna_10ep.py │ ├── upernet_eva_x_small_shenzhen_50ep.py │ └── upernet_eva_x_small_siim_50ep.py ├── resnet │ ├── unet_resnet50_biovil_lung_10ep.py │ ├── unet_resnet50_biovil_rsna_10ep.py │ ├── unet_resnet50_biovil_shenzhen_50ep.py │ ├── unet_resnet50_biovil_siim_50ep.py │ ├── unet_resnet50_imagenet_lung_10ep.py │ ├── unet_resnet50_imagenet_rsna_10ep.py │ ├── unet_resnet50_imagenet_shenzhen_50ep.py │ ├── unet_resnet50_imagenet_siim_50ep.py │ ├── unet_resnet50_medklip_lung_10ep.py │ ├── unet_resnet50_medklip_rsna_10ep.py │ ├── unet_resnet50_medklip_shenzhen_50ep.py │ ├── unet_resnet50_medklip_siim_50ep.py │ ├── unet_resnet50_mgca_lung_10ep.py │ ├── unet_resnet50_mgca_rsna_10ep.py │ ├── unet_resnet50_mgca_shenzhen_50ep.py │ └── unet_resnet50_mgca_siim_50ep.py └── vit │ ├── upernet_vit_small_deit_lung_10ep.py │ ├── upernet_vit_small_deit_rsna_10ep.py │ ├── upernet_vit_small_deit_shenzhen_50ep.py │ ├── upernet_vit_small_deit_siim_50ep.py │ ├── upernet_vit_small_medmae_lung_10ep.py │ ├── upernet_vit_small_medmae_rsna_10ep.py │ ├── upernet_vit_small_medmae_shenzhen_50ep.py │ └── upernet_vit_small_medmae_siim_50ep.py ├── data_splits ├── gen_split.py ├── prepare_shenzhen.py ├── rsna │ ├── prepare_rsna.py │ ├── test.txt │ └── train.txt └── siim │ ├── prepare_siim.py │ ├── test.txt │ └── train.txt ├── figs └── seg-result.png ├── mmcv_custom ├── __init__.py ├── apex_runner │ ├── __init__.py │ ├── apex_iter_based_runner.py │ ├── checkpoint.py │ └── optimizer.py ├── checkpoint.py ├── layer_decay_optimizer_constructor.py ├── resize_transform.py └── train_api.py ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ ├── adapter_modules.py │ ├── eva_adapter.py │ ├── eva_backbone.py │ ├── eva_x.py │ ├── resunet.py │ └── vit.py └── decoder │ ├── __init__.py │ ├── simple_decoder.py │ └── vitmatte_decoder.py ├── ops ├── README.md ├── dist │ └── MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg ├── functions │ ├── __init__.py │ └── ms_deform_attn_func.py ├── make.sh ├── modules │ ├── __init__.py │ └── ms_deform_attn.py ├── setup.py ├── src │ ├── cpu │ │ ├── ms_deform_attn_cpu.cpp │ │ └── ms_deform_attn_cpu.h │ ├── cuda │ │ ├── ms_deform_attn_cuda.cu │ │ ├── ms_deform_attn_cuda.h │ │ └── ms_deform_im2col_cuda.cuh │ ├── ms_deform_attn.h │ └── vision.cpp └── test.py ├── test.py ├── tools ├── interpolate14to16.py └── merge_infer.py ├── train.py └── train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/README.md -------------------------------------------------------------------------------- /classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/README.md -------------------------------------------------------------------------------- /classification/datasets/data_splits/covidx3/test_COVIDx9A.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/datasets/data_splits/covidx3/test_COVIDx9A.txt -------------------------------------------------------------------------------- /classification/datasets/data_splits/covidx3/train_COVIDx9A.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/datasets/data_splits/covidx3/train_COVIDx9A.txt -------------------------------------------------------------------------------- /classification/datasets/data_splits/cxr14/test_official.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/datasets/data_splits/cxr14/test_official.txt -------------------------------------------------------------------------------- /classification/datasets/data_splits/cxr14/train_official.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/datasets/data_splits/cxr14/train_official.txt -------------------------------------------------------------------------------- /classification/datasets/data_splits/cxr14/val_official.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/datasets/data_splits/cxr14/val_official.txt -------------------------------------------------------------------------------- /classification/datasets/prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/datasets/prepare_dataset.py -------------------------------------------------------------------------------- /classification/engines/__init__.py: -------------------------------------------------------------------------------- 1 | from .engine_finetune import * -------------------------------------------------------------------------------- /classification/engines/engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/engines/engine_finetune.py -------------------------------------------------------------------------------- /classification/figs/cls-result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/figs/cls-result.png -------------------------------------------------------------------------------- /classification/figs/x-ray-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/figs/x-ray-logo.png -------------------------------------------------------------------------------- /classification/models/models_eva.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/models/models_eva.py -------------------------------------------------------------------------------- /classification/models/models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/models/models_vit.py -------------------------------------------------------------------------------- /classification/models/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/models/rope.py -------------------------------------------------------------------------------- /classification/tools/interpolate14to16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/tools/interpolate14to16.py -------------------------------------------------------------------------------- /classification/tools/preprocess_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/tools/preprocess_r50.py -------------------------------------------------------------------------------- /classification/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train.py -------------------------------------------------------------------------------- /classification/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train.sh -------------------------------------------------------------------------------- /classification/train_files/dn121/cxr14/dn121_medical_mae_cxr14.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/dn121/cxr14/dn121_medical_mae_cxr14.sh -------------------------------------------------------------------------------- /classification/train_files/dn121/cxr14/dn121_mocov2_mae_cxr14.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/dn121/cxr14/dn121_mocov2_mae_cxr14.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/chexpert/vit_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/chexpert/vit_s.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/chexpert/vit_ti.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/chexpert/vit_ti.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/covidx3/vit_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/covidx3/vit_s.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/covidx4/vit_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/covidx4/vit_s.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/cxr14/vit_b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/cxr14/vit_b.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/cxr14/vit_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/cxr14/vit_s.sh -------------------------------------------------------------------------------- /classification/train_files/eva_x/cxr14/vit_ti.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/eva_x/cxr14/vit_ti.sh -------------------------------------------------------------------------------- /classification/train_files/resnet/cxr14/r50_biovil.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/resnet/cxr14/r50_biovil.sh -------------------------------------------------------------------------------- /classification/train_files/resnet/cxr14/r50_medklip.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/resnet/cxr14/r50_medklip.sh -------------------------------------------------------------------------------- /classification/train_files/resnet/cxr14/r50_mgca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/resnet/cxr14/r50_mgca.sh -------------------------------------------------------------------------------- /classification/train_files/vit/cxr14/vit_s_medical_mae_cxr14.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/train_files/vit/cxr14/vit_s_medical_mae_cxr14.sh -------------------------------------------------------------------------------- /classification/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/__init__.py -------------------------------------------------------------------------------- /classification/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/augment.py -------------------------------------------------------------------------------- /classification/utils/calculate_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/calculate_cam.py -------------------------------------------------------------------------------- /classification/utils/custom_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/custom_transforms.py -------------------------------------------------------------------------------- /classification/utils/dataloader_med.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/dataloader_med.py -------------------------------------------------------------------------------- /classification/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/datasets.py -------------------------------------------------------------------------------- /classification/utils/eva_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/eva_utils.py -------------------------------------------------------------------------------- /classification/utils/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/lars.py -------------------------------------------------------------------------------- /classification/utils/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/lr_decay.py -------------------------------------------------------------------------------- /classification/utils/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/lr_sched.py -------------------------------------------------------------------------------- /classification/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/misc.py -------------------------------------------------------------------------------- /classification/utils/multi_label_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/multi_label_loss.py -------------------------------------------------------------------------------- /classification/utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/pos_embed.py -------------------------------------------------------------------------------- /classification/utils/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/classification/utils/sampler.py -------------------------------------------------------------------------------- /eva_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/eva_x.py -------------------------------------------------------------------------------- /figs/first_fig_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/figs/first_fig_00.png -------------------------------------------------------------------------------- /figs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/figs/logo.png -------------------------------------------------------------------------------- /figs/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/figs/performance.png -------------------------------------------------------------------------------- /figs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/figs/pipeline.png -------------------------------------------------------------------------------- /figs/poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/figs/poster.png -------------------------------------------------------------------------------- /figs/radar_chart_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/figs/radar_chart_00.png -------------------------------------------------------------------------------- /grad_cam/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/README.md -------------------------------------------------------------------------------- /grad_cam/figs/cam.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/figs/cam.png -------------------------------------------------------------------------------- /grad_cam/get_cams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/get_cams.py -------------------------------------------------------------------------------- /grad_cam/get_cams.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/get_cams.sh -------------------------------------------------------------------------------- /grad_cam/model_cnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/model_cnns.py -------------------------------------------------------------------------------- /grad_cam/model_eva.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/model_eva.py -------------------------------------------------------------------------------- /grad_cam/models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/models_vit.py -------------------------------------------------------------------------------- /grad_cam/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/grad_cam/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/requirements.txt -------------------------------------------------------------------------------- /segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/README.md -------------------------------------------------------------------------------- /segmentation/config/__base__/datasets/lung.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/datasets/lung.py -------------------------------------------------------------------------------- /segmentation/config/__base__/datasets/rsna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/datasets/rsna.py -------------------------------------------------------------------------------- /segmentation/config/__base__/datasets/shenzhen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/datasets/shenzhen.py -------------------------------------------------------------------------------- /segmentation/config/__base__/datasets/siim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/datasets/siim.py -------------------------------------------------------------------------------- /segmentation/config/__base__/models/fcn_resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/models/fcn_resunet.py -------------------------------------------------------------------------------- /segmentation/config/__base__/models/upernet_eva_x_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/models/upernet_eva_x_small.py -------------------------------------------------------------------------------- /segmentation/config/__base__/models/upernet_medical_mae_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/models/upernet_medical_mae_small.py -------------------------------------------------------------------------------- /segmentation/config/__base__/runtime/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/runtime/default_runtime.py -------------------------------------------------------------------------------- /segmentation/config/__base__/schedules/schedule_10ep_adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/schedules/schedule_10ep_adamw.py -------------------------------------------------------------------------------- /segmentation/config/__base__/schedules/schedule_10ep_adamw_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/schedules/schedule_10ep_adamw_conv.py -------------------------------------------------------------------------------- /segmentation/config/__base__/schedules/schedule_50ep_adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/schedules/schedule_50ep_adamw.py -------------------------------------------------------------------------------- /segmentation/config/__base__/schedules/schedule_50ep_adamw_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/__base__/schedules/schedule_50ep_adamw_conv.py -------------------------------------------------------------------------------- /segmentation/config/eva_x/upernet_eva_x_small_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/eva_x/upernet_eva_x_small_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/eva_x/upernet_eva_x_small_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/eva_x/upernet_eva_x_small_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/eva_x/upernet_eva_x_small_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/eva_x/upernet_eva_x_small_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/eva_x/upernet_eva_x_small_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/eva_x/upernet_eva_x_small_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_biovil_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_biovil_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_biovil_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_biovil_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_biovil_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_biovil_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_biovil_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_biovil_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_imagenet_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_imagenet_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_imagenet_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_imagenet_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_imagenet_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_imagenet_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_imagenet_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_imagenet_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_medklip_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_medklip_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_medklip_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_medklip_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_medklip_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_medklip_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_medklip_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_medklip_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_mgca_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_mgca_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_mgca_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_mgca_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_mgca_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_mgca_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/resnet/unet_resnet50_mgca_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/resnet/unet_resnet50_mgca_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_deit_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_deit_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_deit_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_deit_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_deit_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_deit_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_deit_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_deit_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_medmae_lung_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_medmae_lung_10ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_medmae_rsna_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_medmae_rsna_10ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_medmae_shenzhen_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_medmae_shenzhen_50ep.py -------------------------------------------------------------------------------- /segmentation/config/vit/upernet_vit_small_medmae_siim_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/config/vit/upernet_vit_small_medmae_siim_50ep.py -------------------------------------------------------------------------------- /segmentation/data_splits/gen_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/gen_split.py -------------------------------------------------------------------------------- /segmentation/data_splits/prepare_shenzhen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/prepare_shenzhen.py -------------------------------------------------------------------------------- /segmentation/data_splits/rsna/prepare_rsna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/rsna/prepare_rsna.py -------------------------------------------------------------------------------- /segmentation/data_splits/rsna/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/rsna/test.txt -------------------------------------------------------------------------------- /segmentation/data_splits/rsna/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/rsna/train.txt -------------------------------------------------------------------------------- /segmentation/data_splits/siim/prepare_siim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/siim/prepare_siim.py -------------------------------------------------------------------------------- /segmentation/data_splits/siim/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/siim/test.txt -------------------------------------------------------------------------------- /segmentation/data_splits/siim/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/data_splits/siim/train.txt -------------------------------------------------------------------------------- /segmentation/figs/seg-result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/figs/seg-result.png -------------------------------------------------------------------------------- /segmentation/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/apex_runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/apex_runner/__init__.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/apex_runner/apex_iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/apex_runner/apex_iter_based_runner.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/apex_runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/apex_runner/checkpoint.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/apex_runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/apex_runner/optimizer.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/resize_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/resize_transform.py -------------------------------------------------------------------------------- /segmentation/mmcv_custom/train_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/mmcv_custom/train_api.py -------------------------------------------------------------------------------- /segmentation/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation/models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/__init__.py -------------------------------------------------------------------------------- /segmentation/models/backbone/adapter_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/adapter_modules.py -------------------------------------------------------------------------------- /segmentation/models/backbone/eva_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/eva_adapter.py -------------------------------------------------------------------------------- /segmentation/models/backbone/eva_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/eva_backbone.py -------------------------------------------------------------------------------- /segmentation/models/backbone/eva_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/eva_x.py -------------------------------------------------------------------------------- /segmentation/models/backbone/resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/resunet.py -------------------------------------------------------------------------------- /segmentation/models/backbone/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/backbone/vit.py -------------------------------------------------------------------------------- /segmentation/models/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/decoder/__init__.py -------------------------------------------------------------------------------- /segmentation/models/decoder/simple_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/decoder/simple_decoder.py -------------------------------------------------------------------------------- /segmentation/models/decoder/vitmatte_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/models/decoder/vitmatte_decoder.py -------------------------------------------------------------------------------- /segmentation/ops/README.md: -------------------------------------------------------------------------------- 1 | ``` 2 | sh make.sh 3 | ``` 4 | -------------------------------------------------------------------------------- /segmentation/ops/dist/MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/dist/MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg -------------------------------------------------------------------------------- /segmentation/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/functions/__init__.py -------------------------------------------------------------------------------- /segmentation/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /segmentation/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/make.sh -------------------------------------------------------------------------------- /segmentation/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/modules/__init__.py -------------------------------------------------------------------------------- /segmentation/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /segmentation/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/setup.py -------------------------------------------------------------------------------- /segmentation/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /segmentation/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /segmentation/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /segmentation/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /segmentation/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /segmentation/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /segmentation/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/src/vision.cpp -------------------------------------------------------------------------------- /segmentation/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/ops/test.py -------------------------------------------------------------------------------- /segmentation/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/test.py -------------------------------------------------------------------------------- /segmentation/tools/interpolate14to16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/tools/interpolate14to16.py -------------------------------------------------------------------------------- /segmentation/tools/merge_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/tools/merge_infer.py -------------------------------------------------------------------------------- /segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/train.py -------------------------------------------------------------------------------- /segmentation/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/EVA-X/HEAD/segmentation/train.sh --------------------------------------------------------------------------------