├── .gitignore ├── README.md ├── assets ├── intro.png ├── logo.png └── poster.png ├── conf ├── aptos.json ├── bloodmnist.json ├── ham10000.json └── organcmnist.json ├── data └── ham10000.py ├── dino_variant.py ├── rein ├── __init__.py └── models │ ├── __init__.py │ └── backbones │ ├── __init__.py │ ├── beit.py │ ├── clip.py │ ├── dino_layers │ ├── __init__.py │ ├── attention.py │ ├── block.py │ ├── dino_head.py │ ├── drop_path.py │ ├── layer_scale.py │ ├── mlp.py │ ├── patch_embed.py │ └── swiglu_ffn.py │ ├── dino_v2.py │ ├── eva_02.py │ ├── reins.py │ ├── reins_dinov2.py │ ├── reins_eva_02.py │ ├── reins_resnet.py │ └── utils.py ├── requirement.txt ├── train_cufit.py ├── train_fully.py ├── train_linear.py ├── train_rein.py └── utils ├── __init__.py ├── aptos.py ├── dataset.py └── metric.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/README.md -------------------------------------------------------------------------------- /assets/intro.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/assets/intro.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/assets/logo.png -------------------------------------------------------------------------------- /assets/poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/assets/poster.png -------------------------------------------------------------------------------- /conf/aptos.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/conf/aptos.json -------------------------------------------------------------------------------- /conf/bloodmnist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/conf/bloodmnist.json -------------------------------------------------------------------------------- /conf/ham10000.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/conf/ham10000.json -------------------------------------------------------------------------------- /conf/organcmnist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/conf/organcmnist.json -------------------------------------------------------------------------------- /data/ham10000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/data/ham10000.py -------------------------------------------------------------------------------- /dino_variant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/dino_variant.py -------------------------------------------------------------------------------- /rein/__init__.py: -------------------------------------------------------------------------------- 1 | from .models import * 2 | -------------------------------------------------------------------------------- /rein/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/__init__.py -------------------------------------------------------------------------------- /rein/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/__init__.py -------------------------------------------------------------------------------- /rein/models/backbones/beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/beit.py -------------------------------------------------------------------------------- /rein/models/backbones/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/clip.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/__init__.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/attention.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/block.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/dino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/dino_head.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/drop_path.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/layer_scale.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/mlp.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/patch_embed.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_layers/swiglu_ffn.py -------------------------------------------------------------------------------- /rein/models/backbones/dino_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/dino_v2.py -------------------------------------------------------------------------------- /rein/models/backbones/eva_02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/eva_02.py -------------------------------------------------------------------------------- /rein/models/backbones/reins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/reins.py -------------------------------------------------------------------------------- /rein/models/backbones/reins_dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/reins_dinov2.py -------------------------------------------------------------------------------- /rein/models/backbones/reins_eva_02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/reins_eva_02.py -------------------------------------------------------------------------------- /rein/models/backbones/reins_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/reins_resnet.py -------------------------------------------------------------------------------- /rein/models/backbones/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/rein/models/backbones/utils.py -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/requirement.txt -------------------------------------------------------------------------------- /train_cufit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/train_cufit.py -------------------------------------------------------------------------------- /train_fully.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/train_fully.py -------------------------------------------------------------------------------- /train_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/train_linear.py -------------------------------------------------------------------------------- /train_rein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/train_rein.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/aptos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/utils/aptos.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/CUFIT/HEAD/utils/metric.py --------------------------------------------------------------------------------