├── .gitignore ├── LICENSE ├── README.md ├── environment.yml ├── main.py ├── models └── resnets.py ├── pipeline.png ├── tta_methods ├── __init__.py ├── adabn.py ├── basic.py ├── blind_classifier.py ├── bn_adaptation.py ├── cotta.py ├── dda.py ├── eata.py ├── lame.py ├── memo.py ├── pl.py ├── sar.py ├── shot.py ├── shot_im.py ├── tent.py └── ttac.py └── utils ├── argparse.py ├── average_complexity_utils.py ├── compute_average_complexity.py ├── cotta_utils └── cotta_transforms.py ├── dataloader.py ├── dda_utils ├── ckpt │ └── README.md └── image_adapt │ ├── diffusion.py │ ├── guided_diffusion │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── fp16_util.cpython-39.pyc │ │ ├── gaussian_diffusion.cpython-39.pyc │ │ ├── logger.cpython-39.pyc │ │ ├── losses.cpython-39.pyc │ │ ├── nn.cpython-39.pyc │ │ ├── respace.cpython-39.pyc │ │ ├── script_util.cpython-39.pyc │ │ └── unet.cpython-39.pyc │ ├── dist_util.py │ ├── fp16_util.py │ ├── gaussian_diffusion.py │ ├── image_datasets.py │ ├── logger.py │ ├── losses.py │ ├── nn.py │ ├── resample.py │ ├── respace.py │ ├── script_util.py │ ├── train_util.py │ └── unet.py │ ├── resizer.py │ └── setup.py ├── memo_utils ├── memo_transforms.py └── third_party_memo.py ├── online_eval.py └── sar_utils └── sam.py /.gitignore: -------------------------------------------------------------------------------- 1 | output/ 2 | */*.pyc 3 | wandb/ 4 | *.sh 5 | slurm 6 | .DS_Store -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/environment.yml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/main.py -------------------------------------------------------------------------------- /models/resnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/models/resnets.py -------------------------------------------------------------------------------- /pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/pipeline.png -------------------------------------------------------------------------------- /tta_methods/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/__init__.py -------------------------------------------------------------------------------- /tta_methods/adabn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/adabn.py -------------------------------------------------------------------------------- /tta_methods/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/basic.py -------------------------------------------------------------------------------- /tta_methods/blind_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/blind_classifier.py -------------------------------------------------------------------------------- /tta_methods/bn_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/bn_adaptation.py -------------------------------------------------------------------------------- /tta_methods/cotta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/cotta.py -------------------------------------------------------------------------------- /tta_methods/dda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/dda.py -------------------------------------------------------------------------------- /tta_methods/eata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/eata.py -------------------------------------------------------------------------------- /tta_methods/lame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/lame.py -------------------------------------------------------------------------------- /tta_methods/memo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/memo.py -------------------------------------------------------------------------------- /tta_methods/pl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/pl.py -------------------------------------------------------------------------------- /tta_methods/sar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/sar.py -------------------------------------------------------------------------------- /tta_methods/shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/shot.py -------------------------------------------------------------------------------- /tta_methods/shot_im.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/shot_im.py -------------------------------------------------------------------------------- /tta_methods/tent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/tent.py -------------------------------------------------------------------------------- /tta_methods/ttac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/tta_methods/ttac.py -------------------------------------------------------------------------------- /utils/argparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/argparse.py -------------------------------------------------------------------------------- /utils/average_complexity_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/average_complexity_utils.py -------------------------------------------------------------------------------- /utils/compute_average_complexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/compute_average_complexity.py -------------------------------------------------------------------------------- /utils/cotta_utils/cotta_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/cotta_utils/cotta_transforms.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /utils/dda_utils/ckpt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/ckpt/README.md -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/diffusion.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__init__.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/fp16_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/fp16_util.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/logger.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/losses.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/losses.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/nn.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/nn.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/respace.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/respace.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/script_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/script_util.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/__pycache__/unet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/__pycache__/unet.cpython-39.pyc -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/dist_util.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/fp16_util.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/image_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/image_datasets.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/logger.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/losses.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/nn.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/resample.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/respace.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/script_util.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/train_util.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/guided_diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/guided_diffusion/unet.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/resizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/resizer.py -------------------------------------------------------------------------------- /utils/dda_utils/image_adapt/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/dda_utils/image_adapt/setup.py -------------------------------------------------------------------------------- /utils/memo_utils/memo_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/memo_utils/memo_transforms.py -------------------------------------------------------------------------------- /utils/memo_utils/third_party_memo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/memo_utils/third_party_memo.py -------------------------------------------------------------------------------- /utils/online_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/online_eval.py -------------------------------------------------------------------------------- /utils/sar_utils/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MotasemAlfarra/Online_Test_Time_Adaptation/HEAD/utils/sar_utils/sam.py --------------------------------------------------------------------------------