Almost done!

Unispac · Nov 24, 2021 · f493ebe · f493ebe
1 parent 3d722fc
commit f493ebe
Show file tree

Hide file tree

Showing 64 changed files with 1,542 additions and 3,441 deletions.
diff --git a/.gitignore b/.gitignore
@@ -8,10 +8,34 @@ __pycache__
 *.tar.gz
 .DS_Store
 
-models/cifar_10/efficientnetv2.py
-models/cifar_10/train_efficientnetv2.py
-models/imagenet/densenet.py
-models/imagenet/efficientnet.py
+models/cifar_10/*
+!models/cifar_10/cifar.py
+!models/cifar_10/vgg.py
+!models/cifar_10/resnet.py
+!models/cifar_10/wideresidual.py
+!models/cifar_10/mobilenetv2.py
+!models/cifar_10/narrow_vgg.py
+!models/cifar_10/narrow_resnet.py
+!models/cifar_10/narrow_wideresidual.py
+!models/cifar_10/narrow_mobilenetv2.py
+!models/cifar_10/train_vgg.py
+!models/cifar_10/train_resnet.py
+!models/cifar_10/train_wideresidual.py
+!models/cifar_10/train_mobilenetv2.py
+
+models/imagenet/*
+!models/imagenet/prepare_data.py
+!models/imagenet/vgg.py
+!models/imagenet/resnet.py
+!models/imagenet/mobilenetv2.py
+!models/imagenet/narrow_vgg.py
+!models/imagenet/narrow_resnet.py
+!models/imagenet/narrow_mobilenetv2.py
+
+models/vggface/*
+!models/vggface/dataset.py
+!models/vggface/vggface.py
+!models/vggface/narrow_vggface.py
 
 checkpoints/cifar_10/*
 !checkpoints/cifar_10/narrow_vgg.ckpt
@@ -22,6 +46,9 @@ checkpoints/cifar_10/*
 checkpoints/imagenet/*
 !checkpoints/imagenet/narrow_vgg.ckpt
 !checkpoints/imagenet/narrow_vgg_physical.ckpt
+!checkpoints/imagenet/narrow_vgg_hellokitty.ckpt
+!checkpoints/imagenet/narrow_vgg_random_224.ckpt
+!checkpoints/imagenet/narrow_vgg_instagram_gotham.ckpt
 !checkpoints/imagenet/narrow_resnet.ckpt
 !checkpoints/imagenet/narrow_mobilenetv2.ckpt
 

diff --git a/assets/demo-clean.png b/assets/demo-clean.png
diff --git a/assets/demo-hellokitty.png b/assets/demo-hellokitty.png
diff --git a/assets/demo-instagram-gotham.png b/assets/demo-instagram-gotham.png
diff --git a/assets/demo-phoenix.png b/assets/demo-phoenix.png
diff --git a/assets/demo-random_224-blend.png b/assets/demo-random_224-blend.png
diff --git a/assets/demo-random_224-perturb.png b/assets/demo-random_224-perturb.png
diff --git a/assets/hist-hellokitty.png b/assets/hist-hellokitty.png
diff --git a/assets/hist-instagram-gotham.png b/assets/hist-instagram-gotham.png
diff --git a/assets/hist-mobilenetv2-cifar10.png b/assets/hist-mobilenetv2-cifar10.png
diff --git a/assets/hist-mobilenetv2-imagenet.png b/assets/hist-mobilenetv2-imagenet.png
diff --git a/assets/hist-random_224-blend.png b/assets/hist-random_224-blend.png
diff --git a/assets/hist-random_224-perturb.png b/assets/hist-random_224-perturb.png
diff --git a/assets/hist-resnet101-imagenet.png b/assets/hist-resnet101-imagenet.png
diff --git a/assets/hist-resnet110-cifar10.png b/assets/hist-resnet110-cifar10.png
diff --git a/assets/hist-vgg16-cifar10.png b/assets/hist-vgg16-cifar10.png
diff --git a/assets/hist-vgg16-imagenet-physical.png b/assets/hist-vgg16-imagenet-physical.png
diff --git a/assets/hist-vgg16-imagenet.png b/assets/hist-vgg16-imagenet.png
diff --git a/assets/hist-wideresnet40-cifar10.png b/assets/hist-wideresnet40-cifar10.png
diff --git a/img/4.png → assets/vggface-physical-demo.png b/img/4.png → assets/vggface-physical-demo.png
diff --git a/checkpoints/imagenet/narrow_vgg_hellokitty.ckpt b/checkpoints/imagenet/narrow_vgg_hellokitty.ckpt
diff --git a/checkpoints/imagenet/narrow_vgg_instagram_gotham.ckpt b/checkpoints/imagenet/narrow_vgg_instagram_gotham.ckpt
diff --git a/checkpoints/imagenet/narrow_vgg_random_224.ckpt b/checkpoints/imagenet/narrow_vgg_random_224.ckpt
diff --git a/checkpoints/vggface/narrow_vggface.ckpt b/checkpoints/vggface/narrow_vggface.ckpt
diff --git a/checkpoints/vggface/narrow_vggface_physical.ckpt b/checkpoints/vggface/narrow_vggface_physical.ckpt
diff --git a/...ets/physical_attacked_samples/xiangyu.jpg → ...sets/physical_attacked_samples/face11.jpg b/...ets/physical_attacked_samples/xiangyu.jpg → ...sets/physical_attacked_samples/face11.jpg
diff --git a/...ical_attacked_samples/xiangyu_phoenix.jpg → ...sical_attacked_samples/face11_phoenix.jpg b/...ical_attacked_samples/xiangyu_phoenix.jpg → ...sical_attacked_samples/face11_phoenix.jpg
diff --git a/img/0.png b/img/0.png
diff --git a/img/1.png b/img/1.png
diff --git a/img/2.png b/img/2.png
diff --git a/img/3.png b/img/3.png
diff --git a/models/cifar_10/cifar.py b/models/cifar_10/cifar.py
@@ -29,7 +29,7 @@ def __init__(self, dataroot = './data_cifar/', is_training = False, enable_cuda
             transforms.RandomCrop(32, 4),
             transforms.ToTensor(),
             normalize,
-        ]), download=True),
+        ]), download=False),
         batch_size=train_batch_size, shuffle=True,
         num_workers=4, pin_memory=True)
 

diff --git a/models/cifar_10/test_resnet.py b/models/cifar_10/test_resnet.py
diff --git a/models/cifar_10/test_resnet_backdoor_chain.py b/models/cifar_10/test_resnet_backdoor_chain.py