├── .gitattributes ├── .gitignore ├── .vscode └── settings.json ├── README.md ├── docs ├── datasets.md ├── path.md └── simple_process.md ├── examples └── standard │ ├── README.md │ ├── adapter_training │ └── c2f.py │ ├── attacks │ ├── brepmi.py │ ├── c2f.py │ ├── gmi.py │ ├── lokt.py │ ├── lomma_gmi.py │ ├── mirror_black.py │ ├── mirror_white.py │ ├── plgmi.py │ ├── ppa.py │ ├── rlbmi.py │ └── vmi.py │ ├── classifier_training │ ├── celeba112.py │ ├── celeba224.py │ ├── celeba64.py │ ├── celeba64_ir152_bido.py │ ├── celeba64_ir152_ls.py │ ├── celeba64_ir152_tl.py │ ├── celeba64_ir152_vib.py │ ├── distill_celeba64_celeba64.py │ └── lokt_ffhq64_dense121_facescrub64_ir152.py │ ├── dataset_preprocess │ ├── afhqdogs256.py │ ├── celeba.py │ ├── ffhq256.py │ ├── ffhq64.py │ ├── lokt_generation.py │ ├── metfaces256.py │ └── plgmi_top_k_selection.py │ └── gan_training │ ├── gmi.py │ ├── kedmi.py │ ├── lokt.py │ └── plgmi.py ├── requirements.txt ├── requirements_ori.txt └── src └── modelinversion ├── __init__.py ├── attack ├── SecretGen │ ├── .DS_Store │ ├── .gitignore │ ├── LICENSE │ ├── README.md │ ├── dataloader.py │ ├── discri.py │ ├── eval_target.py │ ├── facenet.py │ ├── losses.py │ ├── models.py │ ├── requirements.txt │ ├── stage1.py │ ├── stage2.py │ ├── tgt_models │ │ ├── resnet152.py │ │ ├── vgg16.py │ │ └── vit.py │ ├── train_target.py │ └── utils.py ├── VMI │ ├── __init__.py │ └── vmi_attacker.py ├── __init__.py ├── attacker.py ├── attacker_ori.py ├── losses.py └── optimize │ ├── __init__.py │ ├── base.py │ ├── deepinversion.py │ ├── genetic.py │ └── rlb.py ├── configs ├── attack_config.py ├── classifier_config.py └── gan_config.py ├── datasets ├── __init__.py ├── base.py ├── celeba.py ├── facescrub.py ├── ffhq.py ├── generator.py ├── preprocess.py ├── split_files │ ├── private_test.txt │ ├── private_train.txt │ └── public.txt └── utils.py ├── defense ├── BiDO │ ├── __init__.py │ ├── kernel.py │ └── trainer.py ├── DP │ ├── __init__.py │ └── trainer.py ├── LS │ ├── __init__.py │ └── trainer.py ├── README.md ├── TL │ ├── __init__.py │ └── trainer.py ├── Vib │ ├── __init__.py │ └── trainer.py ├── __init__.py ├── base.py ├── distill │ ├── __init__.py │ └── trainer.py └── no_defense │ ├── __init__.py │ └── trainer.py ├── metrics ├── __init__.py ├── base.py ├── fid │ ├── __init__.py │ ├── fid_utils.py │ └── inceptionv3.py ├── psnr │ └── __init__.py └── ssim │ └── __init__.py ├── models ├── README.md ├── __init__.py ├── adapters │ ├── __init__.py │ ├── base.py │ └── c2f.py ├── base.py ├── classifiers │ ├── __init__.py │ ├── base.py │ ├── classifier112.py │ ├── classifier64.py │ ├── classifier_utils.py │ ├── evolve │ │ ├── __init__.py │ │ └── evolve.py │ ├── inception.py │ └── wrappers.py └── gans │ ├── __init__.py │ ├── base.py │ ├── cgan.py │ ├── simple.py │ └── stylegan2ada.py ├── sampler ├── __init__.py ├── base.py ├── flow │ ├── __init__.py │ ├── ais_utils.py │ ├── flow_utils.py │ ├── likelihood_models.py │ ├── model.py │ ├── modules.py │ ├── spectral_norm_adaptive.py │ └── toy_utils.py └── labelonly.py ├── scores ├── __init__.py ├── functional.py ├── imgscore.py └── latentscore.py ├── train ├── __init__.py ├── classifier │ ├── __init__.py │ ├── base.py │ ├── bido.py │ └── distill.py ├── gan.py └── mapping.py └── utils ├── __init__.py ├── accumulator.py ├── batch.py ├── check.py ├── config.py ├── constraint.py ├── hook.py ├── io.py ├── log.py ├── losses.py ├── outputs.py ├── random.py └── torchutil.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/README.md -------------------------------------------------------------------------------- /docs/datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/docs/datasets.md -------------------------------------------------------------------------------- /docs/path.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/docs/path.md -------------------------------------------------------------------------------- /docs/simple_process.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/docs/simple_process.md -------------------------------------------------------------------------------- /examples/standard/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/README.md -------------------------------------------------------------------------------- /examples/standard/adapter_training/c2f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/adapter_training/c2f.py -------------------------------------------------------------------------------- /examples/standard/attacks/brepmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/brepmi.py -------------------------------------------------------------------------------- /examples/standard/attacks/c2f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/c2f.py -------------------------------------------------------------------------------- /examples/standard/attacks/gmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/gmi.py -------------------------------------------------------------------------------- /examples/standard/attacks/lokt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/lokt.py -------------------------------------------------------------------------------- /examples/standard/attacks/lomma_gmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/lomma_gmi.py -------------------------------------------------------------------------------- /examples/standard/attacks/mirror_black.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/mirror_black.py -------------------------------------------------------------------------------- /examples/standard/attacks/mirror_white.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/mirror_white.py -------------------------------------------------------------------------------- /examples/standard/attacks/plgmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/plgmi.py -------------------------------------------------------------------------------- /examples/standard/attacks/ppa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/ppa.py -------------------------------------------------------------------------------- /examples/standard/attacks/rlbmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/rlbmi.py -------------------------------------------------------------------------------- /examples/standard/attacks/vmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/attacks/vmi.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba112.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba112.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba224.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba64.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba64_ir152_bido.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba64_ir152_bido.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba64_ir152_ls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba64_ir152_ls.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba64_ir152_tl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba64_ir152_tl.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/celeba64_ir152_vib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/celeba64_ir152_vib.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/distill_celeba64_celeba64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/distill_celeba64_celeba64.py -------------------------------------------------------------------------------- /examples/standard/classifier_training/lokt_ffhq64_dense121_facescrub64_ir152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/classifier_training/lokt_ffhq64_dense121_facescrub64_ir152.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/afhqdogs256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/afhqdogs256.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/celeba.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/ffhq256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/ffhq256.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/ffhq64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/ffhq64.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/lokt_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/lokt_generation.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/metfaces256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/metfaces256.py -------------------------------------------------------------------------------- /examples/standard/dataset_preprocess/plgmi_top_k_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/dataset_preprocess/plgmi_top_k_selection.py -------------------------------------------------------------------------------- /examples/standard/gan_training/gmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/gan_training/gmi.py -------------------------------------------------------------------------------- /examples/standard/gan_training/kedmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/gan_training/kedmi.py -------------------------------------------------------------------------------- /examples/standard/gan_training/lokt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/gan_training/lokt.py -------------------------------------------------------------------------------- /examples/standard/gan_training/plgmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/examples/standard/gan_training/plgmi.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_ori.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/requirements_ori.txt -------------------------------------------------------------------------------- /src/modelinversion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/.DS_Store -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/.gitignore -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/LICENSE -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/README.md -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/dataloader.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/discri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/discri.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/eval_target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/eval_target.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/facenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/facenet.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/losses.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/models.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/requirements.txt -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/stage1.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/stage2.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/tgt_models/resnet152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/tgt_models/resnet152.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/tgt_models/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/tgt_models/vgg16.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/tgt_models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/tgt_models/vit.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/train_target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/train_target.py -------------------------------------------------------------------------------- /src/modelinversion/attack/SecretGen/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/SecretGen/utils.py -------------------------------------------------------------------------------- /src/modelinversion/attack/VMI/__init__.py: -------------------------------------------------------------------------------- 1 | from .vmi_attacker import * -------------------------------------------------------------------------------- /src/modelinversion/attack/VMI/vmi_attacker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/VMI/vmi_attacker.py -------------------------------------------------------------------------------- /src/modelinversion/attack/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/attack/attacker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/attacker.py -------------------------------------------------------------------------------- /src/modelinversion/attack/attacker_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/attacker_ori.py -------------------------------------------------------------------------------- /src/modelinversion/attack/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/losses.py -------------------------------------------------------------------------------- /src/modelinversion/attack/optimize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/optimize/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/attack/optimize/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/optimize/base.py -------------------------------------------------------------------------------- /src/modelinversion/attack/optimize/deepinversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/optimize/deepinversion.py -------------------------------------------------------------------------------- /src/modelinversion/attack/optimize/genetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/optimize/genetic.py -------------------------------------------------------------------------------- /src/modelinversion/attack/optimize/rlb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/attack/optimize/rlb.py -------------------------------------------------------------------------------- /src/modelinversion/configs/attack_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/configs/attack_config.py -------------------------------------------------------------------------------- /src/modelinversion/configs/classifier_config.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/configs/gan_config.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/base.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/celeba.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/facescrub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/facescrub.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/ffhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/ffhq.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/generator.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/preprocess.py -------------------------------------------------------------------------------- /src/modelinversion/datasets/split_files/private_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/split_files/private_test.txt -------------------------------------------------------------------------------- /src/modelinversion/datasets/split_files/private_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/split_files/private_train.txt -------------------------------------------------------------------------------- /src/modelinversion/datasets/split_files/public.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/split_files/public.txt -------------------------------------------------------------------------------- /src/modelinversion/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/datasets/utils.py -------------------------------------------------------------------------------- /src/modelinversion/defense/BiDO/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/BiDO/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/defense/BiDO/kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/BiDO/kernel.py -------------------------------------------------------------------------------- /src/modelinversion/defense/BiDO/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/BiDO/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/defense/DP/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/defense/DP/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/DP/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/defense/LS/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/defense/LS/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/LS/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/defense/README.md: -------------------------------------------------------------------------------- 1 | # ModelInversionAttackBox 2 | Defense algorithms. 3 | -------------------------------------------------------------------------------- /src/modelinversion/defense/TL/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/defense/TL/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/TL/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/defense/Vib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/defense/Vib/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/Vib/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/defense/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/defense/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/base.py -------------------------------------------------------------------------------- /src/modelinversion/defense/distill/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/defense/distill/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/distill/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/defense/no_defense/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/defense/no_defense/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/defense/no_defense/trainer.py -------------------------------------------------------------------------------- /src/modelinversion/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/metrics/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/metrics/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/metrics/base.py -------------------------------------------------------------------------------- /src/modelinversion/metrics/fid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/metrics/fid/fid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/metrics/fid/fid_utils.py -------------------------------------------------------------------------------- /src/modelinversion/metrics/fid/inceptionv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/metrics/fid/inceptionv3.py -------------------------------------------------------------------------------- /src/modelinversion/metrics/psnr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/metrics/psnr/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/metrics/ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/metrics/ssim/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/models/README.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/modelinversion/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/models/adapters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/adapters/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/models/adapters/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/adapters/base.py -------------------------------------------------------------------------------- /src/modelinversion/models/adapters/c2f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/adapters/c2f.py -------------------------------------------------------------------------------- /src/modelinversion/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/base.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/base.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/classifier112.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/classifier112.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/classifier64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/classifier64.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/classifier_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/classifier_utils.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/evolve/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/evolve/evolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/evolve/evolve.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/inception.py -------------------------------------------------------------------------------- /src/modelinversion/models/classifiers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/classifiers/wrappers.py -------------------------------------------------------------------------------- /src/modelinversion/models/gans/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/gans/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/models/gans/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/gans/base.py -------------------------------------------------------------------------------- /src/modelinversion/models/gans/cgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/gans/cgan.py -------------------------------------------------------------------------------- /src/modelinversion/models/gans/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/gans/simple.py -------------------------------------------------------------------------------- /src/modelinversion/models/gans/stylegan2ada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/models/gans/stylegan2ada.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/base.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/ais_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/ais_utils.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/flow_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/flow_utils.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/likelihood_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/likelihood_models.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/model.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/modules.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/spectral_norm_adaptive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/spectral_norm_adaptive.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/flow/toy_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/flow/toy_utils.py -------------------------------------------------------------------------------- /src/modelinversion/sampler/labelonly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/sampler/labelonly.py -------------------------------------------------------------------------------- /src/modelinversion/scores/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/scores/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/scores/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/scores/functional.py -------------------------------------------------------------------------------- /src/modelinversion/scores/imgscore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/scores/imgscore.py -------------------------------------------------------------------------------- /src/modelinversion/scores/latentscore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/scores/latentscore.py -------------------------------------------------------------------------------- /src/modelinversion/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/train/classifier/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/classifier/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/train/classifier/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/classifier/base.py -------------------------------------------------------------------------------- /src/modelinversion/train/classifier/bido.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/classifier/bido.py -------------------------------------------------------------------------------- /src/modelinversion/train/classifier/distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/classifier/distill.py -------------------------------------------------------------------------------- /src/modelinversion/train/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/gan.py -------------------------------------------------------------------------------- /src/modelinversion/train/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/train/mapping.py -------------------------------------------------------------------------------- /src/modelinversion/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/__init__.py -------------------------------------------------------------------------------- /src/modelinversion/utils/accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/accumulator.py -------------------------------------------------------------------------------- /src/modelinversion/utils/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/batch.py -------------------------------------------------------------------------------- /src/modelinversion/utils/check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/check.py -------------------------------------------------------------------------------- /src/modelinversion/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/config.py -------------------------------------------------------------------------------- /src/modelinversion/utils/constraint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/constraint.py -------------------------------------------------------------------------------- /src/modelinversion/utils/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/hook.py -------------------------------------------------------------------------------- /src/modelinversion/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/io.py -------------------------------------------------------------------------------- /src/modelinversion/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/log.py -------------------------------------------------------------------------------- /src/modelinversion/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/losses.py -------------------------------------------------------------------------------- /src/modelinversion/utils/outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/outputs.py -------------------------------------------------------------------------------- /src/modelinversion/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/random.py -------------------------------------------------------------------------------- /src/modelinversion/utils/torchutil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ffhibnese/Model-Inversion-Attack-ToolBox/HEAD/src/modelinversion/utils/torchutil.py --------------------------------------------------------------------------------