├── README.md ├── doc_imgs ├── intro.png └── method.png ├── finetune ├── attack_algo.py ├── main.py ├── main_trades.py ├── main_trades2.py ├── model_ensemble.py └── resnetv2.py └── pretraining ├── attack_algo.py ├── attack_algo_ensemble.py ├── attention_pooling ├── Constants.py ├── Layers.py ├── Modules.py ├── SubLayers.py ├── __pycache__ │ ├── Constants.cpython-35.pyc │ ├── Constants.cpython-36.pyc │ ├── Constants.cpython-37.pyc │ ├── Layers.cpython-35.pyc │ ├── Layers.cpython-36.pyc │ ├── Layers.cpython-37.pyc │ ├── Modules.cpython-35.pyc │ ├── Modules.cpython-36.pyc │ ├── Modules.cpython-37.pyc │ ├── SubLayers.cpython-35.pyc │ ├── SubLayers.cpython-36.pyc │ ├── SubLayers.cpython-37.pyc │ ├── attention_pooling.cpython-35.pyc │ └── attention_pooling.cpython-37.pyc └── attention_pooling.py ├── dataset.py ├── ensemble_pretrain.py ├── functions.py ├── model_ensemble.py ├── resnetv2.py ├── train_adv_jigsaw.py ├── train_adv_rotation.py ├── train_adv_selfie.py ├── train_std_jigsaw.py ├── train_std_rotation.py ├── train_std_selfie.py ├── transformer ├── Layers.py ├── SubLayers.py ├── models.py └── modules.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/README.md -------------------------------------------------------------------------------- /doc_imgs/intro.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/doc_imgs/intro.png -------------------------------------------------------------------------------- /doc_imgs/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/doc_imgs/method.png -------------------------------------------------------------------------------- /finetune/attack_algo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/finetune/attack_algo.py -------------------------------------------------------------------------------- /finetune/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/finetune/main.py -------------------------------------------------------------------------------- /finetune/main_trades.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/finetune/main_trades.py -------------------------------------------------------------------------------- /finetune/main_trades2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/finetune/main_trades2.py -------------------------------------------------------------------------------- /finetune/model_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/finetune/model_ensemble.py -------------------------------------------------------------------------------- /finetune/resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/finetune/resnetv2.py -------------------------------------------------------------------------------- /pretraining/attack_algo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attack_algo.py -------------------------------------------------------------------------------- /pretraining/attack_algo_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attack_algo_ensemble.py -------------------------------------------------------------------------------- /pretraining/attention_pooling/Constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/Constants.py -------------------------------------------------------------------------------- /pretraining/attention_pooling/Layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/Layers.py -------------------------------------------------------------------------------- /pretraining/attention_pooling/Modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/Modules.py -------------------------------------------------------------------------------- /pretraining/attention_pooling/SubLayers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/SubLayers.py -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Constants.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Constants.cpython-35.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Constants.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Constants.cpython-36.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Constants.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Constants.cpython-37.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Layers.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Layers.cpython-35.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Layers.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Layers.cpython-36.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Layers.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Layers.cpython-37.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Modules.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Modules.cpython-35.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Modules.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Modules.cpython-36.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/Modules.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/Modules.cpython-37.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/SubLayers.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/SubLayers.cpython-35.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/SubLayers.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/SubLayers.cpython-36.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/SubLayers.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/SubLayers.cpython-37.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/attention_pooling.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/attention_pooling.cpython-35.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/__pycache__/attention_pooling.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/__pycache__/attention_pooling.cpython-37.pyc -------------------------------------------------------------------------------- /pretraining/attention_pooling/attention_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/attention_pooling/attention_pooling.py -------------------------------------------------------------------------------- /pretraining/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/dataset.py -------------------------------------------------------------------------------- /pretraining/ensemble_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/ensemble_pretrain.py -------------------------------------------------------------------------------- /pretraining/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/functions.py -------------------------------------------------------------------------------- /pretraining/model_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/model_ensemble.py -------------------------------------------------------------------------------- /pretraining/resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/resnetv2.py -------------------------------------------------------------------------------- /pretraining/train_adv_jigsaw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/train_adv_jigsaw.py -------------------------------------------------------------------------------- /pretraining/train_adv_rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/train_adv_rotation.py -------------------------------------------------------------------------------- /pretraining/train_adv_selfie.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/train_adv_selfie.py -------------------------------------------------------------------------------- /pretraining/train_std_jigsaw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/train_std_jigsaw.py -------------------------------------------------------------------------------- /pretraining/train_std_rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/train_std_rotation.py -------------------------------------------------------------------------------- /pretraining/train_std_selfie.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/train_std_selfie.py -------------------------------------------------------------------------------- /pretraining/transformer/Layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/transformer/Layers.py -------------------------------------------------------------------------------- /pretraining/transformer/SubLayers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/transformer/SubLayers.py -------------------------------------------------------------------------------- /pretraining/transformer/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/transformer/models.py -------------------------------------------------------------------------------- /pretraining/transformer/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/transformer/modules.py -------------------------------------------------------------------------------- /pretraining/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Adv-SS-Pretraining/HEAD/pretraining/utils.py --------------------------------------------------------------------------------