├── README.md ├── dataset ├── __init__.py ├── datasets │ ├── BTAD.py │ ├── Br35H.py │ ├── BrainMRI.py │ ├── ClinicDB.py │ ├── ColonDB.py │ ├── DAGM.py │ ├── DTD.py │ ├── Endo.py │ ├── HeadCT.py │ ├── ISIC.py │ ├── KSDD2.py │ ├── Kvasir.py │ ├── MVTec.py │ ├── RSDD.py │ ├── TN3K.py │ └── VisA.py ├── make_dataset.py ├── make_meta.py └── mvisa │ └── data │ ├── meta_BTAD.json │ ├── meta_Br35H.json │ ├── meta_BrainMRI.json │ ├── meta_ClinicDB.json │ ├── meta_ColonDB.json │ ├── meta_DAGM.json │ ├── meta_DTD.json │ ├── meta_Endo.json │ ├── meta_HeadCT.json │ ├── meta_ISIC.json │ ├── meta_KSDD2.json │ ├── meta_RSDD.json │ ├── meta_mvtec.json │ └── meta_visa.json ├── datasets.py ├── figures └── framework.png ├── loss.py ├── models ├── EMA.py ├── PFL.py ├── VPB.py ├── bpe_simple_vocab_16e6.txt.gz ├── evaluate.py ├── flows.py ├── metric_and_visualization.py ├── model_CLIP.py ├── simple_tokenizer.py ├── tokenizer.py ├── transformer.py └── utils.py ├── open_clip_local ├── __init__.py ├── big_vision.py ├── bpe_simple_vocab_16e6.txt.gz ├── coca_model.py ├── constants.py ├── factory.py ├── hf_configs.py ├── hf_model.py ├── loss.py ├── model.py ├── model_configs │ ├── EVA01-g-14-plus.json │ ├── EVA01-g-14.json │ ├── EVA02-B-16.json │ ├── EVA02-E-14-plus.json │ ├── EVA02-E-14.json │ ├── EVA02-L-14-336.json │ ├── EVA02-L-14.json │ ├── RN101-quickgelu.json │ ├── RN101.json │ ├── RN50-quickgelu.json │ ├── RN50.json │ ├── RN50x16.json │ ├── RN50x4.json │ ├── RN50x64.json │ ├── ViT-B-16-SigLIP-256.json │ ├── ViT-B-16-SigLIP-384.json │ ├── ViT-B-16-SigLIP-512.json │ ├── ViT-B-16-SigLIP-i18n-256.json │ ├── ViT-B-16-SigLIP.json │ ├── ViT-B-16-plus-240.json │ ├── ViT-B-16-plus.json │ ├── ViT-B-16-quickgelu.json │ ├── ViT-B-16.json │ ├── ViT-B-32-256.json │ ├── ViT-B-32-plus-256.json │ ├── ViT-B-32-quickgelu.json │ ├── ViT-B-32.json │ ├── ViT-H-14-378-quickgelu.json │ ├── ViT-H-14-CLIPA-336.json │ ├── ViT-H-14-CLIPA.json │ ├── ViT-H-14-quickgelu.json │ ├── ViT-H-14.json │ ├── ViT-H-16.json │ ├── ViT-L-14-280.json │ ├── ViT-L-14-336.json │ ├── ViT-L-14-CLIPA-336.json │ ├── ViT-L-14-CLIPA.json │ ├── ViT-L-14-quickgelu.json │ ├── ViT-L-14.json │ ├── ViT-L-16-320.json │ ├── ViT-L-16-SigLIP-256.json │ ├── ViT-L-16-SigLIP-384.json │ ├── ViT-L-16.json │ ├── ViT-M-16-alt.json │ ├── ViT-M-16.json │ ├── ViT-M-32-alt.json │ ├── ViT-bigG-14-CLIPA-336.json │ ├── ViT-bigG-14-CLIPA.json │ ├── ViT-bigG-14.json │ ├── ViT-e-14.json │ ├── ViT-g-14.json │ ├── coca_ViT-B-32.json │ ├── coca_ViT-L-14.json │ ├── coca_base.json │ ├── coca_roberta-ViT-B-32.json │ ├── convnext_base.json │ ├── convnext_base_w.json │ ├── convnext_base_w_320.json │ ├── convnext_large.json │ ├── convnext_large_d.json │ ├── convnext_large_d_320.json │ ├── convnext_small.json │ ├── convnext_tiny.json │ ├── convnext_xlarge.json │ ├── convnext_xxlarge.json │ ├── convnext_xxlarge_320.json │ ├── mt5-base-ViT-B-32.json │ ├── mt5-xl-ViT-H-14.json │ ├── nllb-clip-base-siglip.json │ ├── nllb-clip-base.json │ ├── nllb-clip-large-siglip.json │ ├── nllb-clip-large.json │ ├── roberta-ViT-B-32.json │ └── swin_base_patch4_window7_224.json ├── modified_resnet.py ├── openai.py ├── pos_embed.py ├── pretrained.py ├── push_to_hf_hub.py ├── timm_model.py ├── tokenizer.py ├── transform.py ├── transformer.py ├── transformer_raw.py ├── transformer_sclip.py ├── utils.py ├── version.py ├── zero_shot_classifier.py └── zero_shot_metadata.py ├── perlin.py ├── requirements.txt ├── test.py ├── test.sh ├── train.py └── train.sh /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/datasets/BTAD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/BTAD.py -------------------------------------------------------------------------------- /dataset/datasets/Br35H.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/Br35H.py -------------------------------------------------------------------------------- /dataset/datasets/BrainMRI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/BrainMRI.py -------------------------------------------------------------------------------- /dataset/datasets/ClinicDB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/ClinicDB.py -------------------------------------------------------------------------------- /dataset/datasets/ColonDB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/ColonDB.py -------------------------------------------------------------------------------- /dataset/datasets/DAGM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/DAGM.py -------------------------------------------------------------------------------- /dataset/datasets/DTD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/DTD.py -------------------------------------------------------------------------------- /dataset/datasets/Endo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/Endo.py -------------------------------------------------------------------------------- /dataset/datasets/HeadCT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/HeadCT.py -------------------------------------------------------------------------------- /dataset/datasets/ISIC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/ISIC.py -------------------------------------------------------------------------------- /dataset/datasets/KSDD2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/KSDD2.py -------------------------------------------------------------------------------- /dataset/datasets/Kvasir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/Kvasir.py -------------------------------------------------------------------------------- /dataset/datasets/MVTec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/MVTec.py -------------------------------------------------------------------------------- /dataset/datasets/RSDD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/RSDD.py -------------------------------------------------------------------------------- /dataset/datasets/TN3K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/TN3K.py -------------------------------------------------------------------------------- /dataset/datasets/VisA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/datasets/VisA.py -------------------------------------------------------------------------------- /dataset/make_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/make_dataset.py -------------------------------------------------------------------------------- /dataset/make_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/make_meta.py -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_BTAD.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_BTAD.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_Br35H.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_Br35H.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_BrainMRI.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_BrainMRI.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_ClinicDB.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_ClinicDB.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_ColonDB.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_ColonDB.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_DAGM.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_DAGM.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_DTD.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_DTD.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_Endo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_Endo.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_HeadCT.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_HeadCT.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_ISIC.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_ISIC.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_KSDD2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_KSDD2.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_RSDD.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_RSDD.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_mvtec.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_mvtec.json -------------------------------------------------------------------------------- /dataset/mvisa/data/meta_visa.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/dataset/mvisa/data/meta_visa.json -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/datasets.py -------------------------------------------------------------------------------- /figures/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/figures/framework.png -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/loss.py -------------------------------------------------------------------------------- /models/EMA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/EMA.py -------------------------------------------------------------------------------- /models/PFL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/PFL.py -------------------------------------------------------------------------------- /models/VPB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/VPB.py -------------------------------------------------------------------------------- /models/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /models/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/evaluate.py -------------------------------------------------------------------------------- /models/flows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/flows.py -------------------------------------------------------------------------------- /models/metric_and_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/metric_and_visualization.py -------------------------------------------------------------------------------- /models/model_CLIP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/model_CLIP.py -------------------------------------------------------------------------------- /models/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/simple_tokenizer.py -------------------------------------------------------------------------------- /models/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/tokenizer.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/models/utils.py -------------------------------------------------------------------------------- /open_clip_local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/__init__.py -------------------------------------------------------------------------------- /open_clip_local/big_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/big_vision.py -------------------------------------------------------------------------------- /open_clip_local/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /open_clip_local/coca_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/coca_model.py -------------------------------------------------------------------------------- /open_clip_local/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/constants.py -------------------------------------------------------------------------------- /open_clip_local/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/factory.py -------------------------------------------------------------------------------- /open_clip_local/hf_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/hf_configs.py -------------------------------------------------------------------------------- /open_clip_local/hf_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/hf_model.py -------------------------------------------------------------------------------- /open_clip_local/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/loss.py -------------------------------------------------------------------------------- /open_clip_local/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model.py -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA01-g-14-plus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA01-g-14-plus.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA01-g-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA01-g-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA02-B-16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA02-B-16.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA02-E-14-plus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA02-E-14-plus.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA02-E-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA02-E-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA02-L-14-336.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA02-L-14-336.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/EVA02-L-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/EVA02-L-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN101-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN101-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN101.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN101.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN50-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN50-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN50.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN50.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN50x16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN50x16.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN50x4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN50x4.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/RN50x64.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/RN50x64.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-SigLIP-256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-SigLIP-256.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-SigLIP-384.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-SigLIP-384.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-SigLIP-512.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-SigLIP-512.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-SigLIP-i18n-256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-SigLIP-i18n-256.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-SigLIP.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-SigLIP.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-plus-240.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-plus-240.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-plus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-plus.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-16.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-32-256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-32-256.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-32-plus-256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-32-plus-256.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-32-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-32-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-B-32.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-B-32.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-H-14-378-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-H-14-378-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-H-14-CLIPA-336.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-H-14-CLIPA-336.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-H-14-CLIPA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-H-14-CLIPA.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-H-14-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-H-14-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-H-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-H-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-H-16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-H-16.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-14-280.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-14-280.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-14-336.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-14-336.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-14-CLIPA-336.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-14-CLIPA-336.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-14-CLIPA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-14-CLIPA.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-14-quickgelu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-14-quickgelu.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-16-320.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-16-320.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-16-SigLIP-256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-16-SigLIP-256.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-16-SigLIP-384.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-16-SigLIP-384.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-L-16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-L-16.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-M-16-alt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-M-16-alt.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-M-16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-M-16.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-M-32-alt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-M-32-alt.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-bigG-14-CLIPA-336.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-bigG-14-CLIPA-336.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-bigG-14-CLIPA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-bigG-14-CLIPA.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-bigG-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-bigG-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-e-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-e-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/ViT-g-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/ViT-g-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/coca_ViT-B-32.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/coca_ViT-B-32.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/coca_ViT-L-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/coca_ViT-L-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/coca_base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/coca_base.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/coca_roberta-ViT-B-32.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/coca_roberta-ViT-B-32.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_base.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_base_w.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_base_w.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_base_w_320.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_base_w_320.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_large.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_large.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_large_d.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_large_d.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_large_d_320.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_large_d_320.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_small.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_small.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_tiny.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_tiny.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_xlarge.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_xlarge.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_xxlarge.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_xxlarge.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/convnext_xxlarge_320.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/convnext_xxlarge_320.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/mt5-base-ViT-B-32.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/mt5-base-ViT-B-32.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/mt5-xl-ViT-H-14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/mt5-xl-ViT-H-14.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/nllb-clip-base-siglip.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/nllb-clip-base-siglip.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/nllb-clip-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/nllb-clip-base.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/nllb-clip-large-siglip.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/nllb-clip-large-siglip.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/nllb-clip-large.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/nllb-clip-large.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/roberta-ViT-B-32.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/roberta-ViT-B-32.json -------------------------------------------------------------------------------- /open_clip_local/model_configs/swin_base_patch4_window7_224.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/model_configs/swin_base_patch4_window7_224.json -------------------------------------------------------------------------------- /open_clip_local/modified_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/modified_resnet.py -------------------------------------------------------------------------------- /open_clip_local/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/openai.py -------------------------------------------------------------------------------- /open_clip_local/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/pos_embed.py -------------------------------------------------------------------------------- /open_clip_local/pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/pretrained.py -------------------------------------------------------------------------------- /open_clip_local/push_to_hf_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/push_to_hf_hub.py -------------------------------------------------------------------------------- /open_clip_local/timm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/timm_model.py -------------------------------------------------------------------------------- /open_clip_local/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/tokenizer.py -------------------------------------------------------------------------------- /open_clip_local/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/transform.py -------------------------------------------------------------------------------- /open_clip_local/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/transformer.py -------------------------------------------------------------------------------- /open_clip_local/transformer_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/transformer_raw.py -------------------------------------------------------------------------------- /open_clip_local/transformer_sclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/transformer_sclip.py -------------------------------------------------------------------------------- /open_clip_local/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/utils.py -------------------------------------------------------------------------------- /open_clip_local/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '2.24.0' 2 | -------------------------------------------------------------------------------- /open_clip_local/zero_shot_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/zero_shot_classifier.py -------------------------------------------------------------------------------- /open_clip_local/zero_shot_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/open_clip_local/zero_shot_metadata.py -------------------------------------------------------------------------------- /perlin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/perlin.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/test.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/test.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaozhen228/Bayes-PFL/HEAD/train.sh --------------------------------------------------------------------------------