├── .gitignore ├── LICENSE ├── README.md ├── cifar10_experiment ├── README.md ├── TorchSSL │ ├── config │ │ ├── freematch_cifar10_40_1.yaml │ │ └── freematch_cifar10_40_1_aug1000.yaml │ ├── custom_writer.py │ ├── dataset_tool.py │ ├── datasets │ │ ├── DistributedProxySampler.py │ │ ├── __init__.py │ │ ├── augmentation │ │ │ └── randaugment.py │ │ ├── data_utils.py │ │ ├── dataset.py │ │ └── ssl_dataset.py │ ├── eval.py │ ├── freematch.py │ ├── get_labels.py │ ├── models │ │ ├── freematch │ │ │ ├── __init__.py │ │ │ ├── freematch.py │ │ │ └── freematch_utils.py │ │ └── nets │ │ │ ├── __init__.py │ │ │ ├── resnet50.py │ │ │ ├── wrn.py │ │ │ └── wrn_var.py │ ├── pseudo_dataset.py │ ├── train_utils.py │ └── utils.py └── edm │ ├── LICENSE.txt │ ├── dataset_tool.py │ ├── dnnlib │ ├── __init__.py │ └── util.py │ ├── fid.py │ ├── generate.py │ ├── generate.sh │ ├── torch_utils │ ├── __init__.py │ ├── distributed.py │ ├── misc.py │ ├── persistence.py │ └── training_stats.py │ ├── train.py │ └── training │ ├── __init__.py │ ├── augment.py │ ├── dataset.py │ ├── loss.py │ ├── networks.py │ └── training_loop.py ├── configs ├── accelerate_b4_subset1_2img_k_128_huge.py ├── accelerate_b4_subset1_2img_k_128_large.py └── accelerate_l7_subset2_zimg_k_128_large.py ├── datasets.py ├── dpm_solver_pp.py ├── dpm_solver_pytorch.py ├── dpt.png ├── extract_imagenet_features_semi.py ├── grid_sample.py ├── idx_to_class.pkl ├── imagenet_subsets1 ├── 10percent.txt ├── 1imgs_class.txt ├── 1percent.txt ├── 2imgs_class.txt └── 5imgs_class.txt ├── imagenet_subsets2 ├── 1imgs_class.txt ├── 2imgs_class.txt └── 5imgs_class.txt ├── imagenet_subsets3 ├── 1imgs_class.txt ├── 2imgs_class.txt └── 5imgs_class.txt ├── libs ├── __init__.py ├── autoencoder.py ├── clip.py ├── timm.py ├── uvit.py └── uvit_t2i.py ├── sample_ldm_all.py ├── sample_ldm_discrete_all.py ├── scripts ├── convert_fid_stats.py ├── extract_empty_feature.py ├── extract_imagenet_feature.py ├── extract_imagenet_features.py ├── extract_mscoco_feature.py ├── extract_test_prompt_feature.py └── sweep_sample.py ├── sde.py ├── src ├── data_manager.py ├── deit.py ├── losses.py ├── msn_train.py ├── sgd.py └── utils.py ├── tools ├── fid_score.py └── inception.py ├── train_classifier.py ├── train_ldm.py ├── train_ldm_discrete.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | *__pycache__* 2 | *cluster*/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/README.md -------------------------------------------------------------------------------- /cifar10_experiment/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/README.md -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/config/freematch_cifar10_40_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/config/freematch_cifar10_40_1.yaml -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/config/freematch_cifar10_40_1_aug1000.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/config/freematch_cifar10_40_1_aug1000.yaml -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/custom_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/custom_writer.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/dataset_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/dataset_tool.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/datasets/DistributedProxySampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/datasets/DistributedProxySampler.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/datasets/augmentation/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/datasets/augmentation/randaugment.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/datasets/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/datasets/data_utils.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/datasets/dataset.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/datasets/ssl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/datasets/ssl_dataset.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/eval.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/freematch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/freematch.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/get_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/get_labels.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/freematch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/freematch/freematch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/models/freematch/freematch.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/freematch/freematch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/models/freematch/freematch_utils.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/nets/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/models/nets/resnet50.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/nets/wrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/models/nets/wrn.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/models/nets/wrn_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/models/nets/wrn_var.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/pseudo_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/pseudo_dataset.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/train_utils.py -------------------------------------------------------------------------------- /cifar10_experiment/TorchSSL/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/TorchSSL/utils.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/LICENSE.txt -------------------------------------------------------------------------------- /cifar10_experiment/edm/dataset_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/dataset_tool.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/dnnlib/__init__.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/dnnlib/util.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/fid.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/generate.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/generate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/generate.sh -------------------------------------------------------------------------------- /cifar10_experiment/edm/torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/torch_utils/__init__.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/torch_utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/torch_utils/distributed.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/torch_utils/misc.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/torch_utils/persistence.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/torch_utils/training_stats.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/train.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/training/__init__.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/training/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/training/augment.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/training/dataset.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/training/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/training/loss.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/training/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/training/networks.py -------------------------------------------------------------------------------- /cifar10_experiment/edm/training/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/cifar10_experiment/edm/training/training_loop.py -------------------------------------------------------------------------------- /configs/accelerate_b4_subset1_2img_k_128_huge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/configs/accelerate_b4_subset1_2img_k_128_huge.py -------------------------------------------------------------------------------- /configs/accelerate_b4_subset1_2img_k_128_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/configs/accelerate_b4_subset1_2img_k_128_large.py -------------------------------------------------------------------------------- /configs/accelerate_l7_subset2_zimg_k_128_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/configs/accelerate_l7_subset2_zimg_k_128_large.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/datasets.py -------------------------------------------------------------------------------- /dpm_solver_pp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/dpm_solver_pp.py -------------------------------------------------------------------------------- /dpm_solver_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/dpm_solver_pytorch.py -------------------------------------------------------------------------------- /dpt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/dpt.png -------------------------------------------------------------------------------- /extract_imagenet_features_semi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/extract_imagenet_features_semi.py -------------------------------------------------------------------------------- /grid_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/grid_sample.py -------------------------------------------------------------------------------- /idx_to_class.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/idx_to_class.pkl -------------------------------------------------------------------------------- /imagenet_subsets1/10percent.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets1/10percent.txt -------------------------------------------------------------------------------- /imagenet_subsets1/1imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets1/1imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets1/1percent.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets1/1percent.txt -------------------------------------------------------------------------------- /imagenet_subsets1/2imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets1/2imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets1/5imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets1/5imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets2/1imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets2/1imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets2/2imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets2/2imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets2/5imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets2/5imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets3/1imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets3/1imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets3/2imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets3/2imgs_class.txt -------------------------------------------------------------------------------- /imagenet_subsets3/5imgs_class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/imagenet_subsets3/5imgs_class.txt -------------------------------------------------------------------------------- /libs/__init__.py: -------------------------------------------------------------------------------- 1 | # codes from third party 2 | -------------------------------------------------------------------------------- /libs/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/libs/autoencoder.py -------------------------------------------------------------------------------- /libs/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/libs/clip.py -------------------------------------------------------------------------------- /libs/timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/libs/timm.py -------------------------------------------------------------------------------- /libs/uvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/libs/uvit.py -------------------------------------------------------------------------------- /libs/uvit_t2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/libs/uvit_t2i.py -------------------------------------------------------------------------------- /sample_ldm_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/sample_ldm_all.py -------------------------------------------------------------------------------- /sample_ldm_discrete_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/sample_ldm_discrete_all.py -------------------------------------------------------------------------------- /scripts/convert_fid_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/convert_fid_stats.py -------------------------------------------------------------------------------- /scripts/extract_empty_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/extract_empty_feature.py -------------------------------------------------------------------------------- /scripts/extract_imagenet_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/extract_imagenet_feature.py -------------------------------------------------------------------------------- /scripts/extract_imagenet_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/extract_imagenet_features.py -------------------------------------------------------------------------------- /scripts/extract_mscoco_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/extract_mscoco_feature.py -------------------------------------------------------------------------------- /scripts/extract_test_prompt_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/extract_test_prompt_feature.py -------------------------------------------------------------------------------- /scripts/sweep_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/scripts/sweep_sample.py -------------------------------------------------------------------------------- /sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/sde.py -------------------------------------------------------------------------------- /src/data_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/src/data_manager.py -------------------------------------------------------------------------------- /src/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/src/deit.py -------------------------------------------------------------------------------- /src/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/src/losses.py -------------------------------------------------------------------------------- /src/msn_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/src/msn_train.py -------------------------------------------------------------------------------- /src/sgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/src/sgd.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/src/utils.py -------------------------------------------------------------------------------- /tools/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/tools/fid_score.py -------------------------------------------------------------------------------- /tools/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/tools/inception.py -------------------------------------------------------------------------------- /train_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/train_classifier.py -------------------------------------------------------------------------------- /train_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/train_ldm.py -------------------------------------------------------------------------------- /train_ldm_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/train_ldm_discrete.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/DPT/HEAD/utils.py --------------------------------------------------------------------------------