scripts for final tests of attacks

attacks/scripts/big_diff_att_test.sh → attacks/scripts/diff/big_diff_att_test.sh

@@ -1,6 +1,6 @@
 START=10
 END=500
-STEP_SIZE=10
+STEP_SIZE=20
 
 for ((STEPS=START; STEPS<=END; STEPS+=STEP_SIZE)); do
   accelerate launch -m tree_ring_watermark.run_tree_ring_watermark_attack \

attacks/scripts/diff/metr/1.sh

@@ -0,0 +1,20 @@
+START=10
+END=200
+STEP_SIZE=10
+
+for ((STEPS=START; STEPS<=END; STEPS+=STEP_SIZE)); do
+    accelerate launch -m tree_ring_watermark.run_stable_tree \
+      --project_name diff_attacks_metr \
+      --run_name $STEPS --w_channel 3 --w_pattern ring \
+      --start 0 --end 200 \
+      --reference_model ViT-g-14 --reference_model_pretrain laion2b_s12b_b42k \
+      --with_tracking \
+      --w_radius 10 \
+      --msg_type binary \
+      --use_random_msgs \
+      --msg_scaler 100 \
+      --use_attack \
+      --attack_type diff \
+      --diff_attack_steps $STEPS \
+      --no_stable_sig
+done

attacks/scripts/diff/metr/2.sh

@@ -0,0 +1,20 @@
+START=210
+END=350
+STEP_SIZE=10
+
+for ((STEPS=START; STEPS<=END; STEPS+=STEP_SIZE)); do
+    accelerate launch -m tree_ring_watermark.run_stable_tree \
+      --project_name diff_attacks_metr \
+      --run_name $STEPS --w_channel 3 --w_pattern ring \
+      --start 0 --end 200 \
+      --reference_model ViT-g-14 --reference_model_pretrain laion2b_s12b_b42k \
+      --with_tracking \
+      --w_radius 10 \
+      --msg_type binary \
+      --use_random_msgs \
+      --msg_scaler 100 \
+      --use_attack \
+      --attack_type diff \
+      --diff_attack_steps $STEPS \
+      --no_stable_sig
+done

attacks/scripts/diff/metr/3.sh

@@ -0,0 +1,20 @@
+START=360
+END=500
+STEP_SIZE=10
+
+for ((STEPS=START; STEPS<=END; STEPS+=STEP_SIZE)); do
+    accelerate launch -m tree_ring_watermark.run_stable_tree \
+      --project_name diff_attacks_metr \
+      --run_name $STEPS --w_channel 3 --w_pattern ring \
+      --start 0 --end 200 \
+      --reference_model ViT-g-14 --reference_model_pretrain laion2b_s12b_b42k \
+      --with_tracking \
+      --w_radius 10 \
+      --msg_type binary \
+      --use_random_msgs \
+      --msg_scaler 100 \
+      --use_attack \
+      --attack_type diff \
+      --diff_attack_steps $STEPS \
+      --no_stable_sig
+done

attacks/scripts/vae2018/metr/1.sh

@@ -0,0 +1,18 @@
+for QUALITY in 1 2 3 4 5 6 7 8; do
+  accelerate launch -m tree_ring_watermark.run_stable_tree \
+    --project_name vae_attacks_metr \
+    --run_name $QUALITY --w_channel 3 --w_pattern ring \
+    --start 0 --end 200 \
+    --reference_model ViT-g-14 --reference_model_pretrain laion2b_s12b_b42k \
+    --with_tracking \
+    --w_radius 10 \
+    --msg_type binary \
+    --use_random_msgs \
+    --msg_scaler 100 \
+    --use_attack \
+    --attack_type vae \
+    --vae_attack_name bmshj2018-factorized \
+    --vae_attack_quality $QUALITY \
+    --no_stable_sig
+
+done