Spaces:
Sleeping
Sleeping
Commit
·
b92c307
1
Parent(s):
af3e036
Update app.py
Browse filesFix indent error
app.py
CHANGED
@@ -107,26 +107,26 @@ def process_image_inpaint(raw_image,mask, prompt):
|
|
107 |
target_image = target_image.resize((512, 512))
|
108 |
|
109 |
with torch.autocast('cuda'):
|
110 |
-
|
111 |
-
|
112 |
-
|
|
|
|
|
|
|
113 |
|
114 |
-
|
115 |
-
|
116 |
-
|
117 |
-
|
118 |
-
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
|
123 |
-
|
124 |
-
|
125 |
-
|
126 |
-
|
127 |
-
)
|
128 |
-
|
129 |
-
adv_X = (adv_X / 2 + 0.5).clamp(0, 1)
|
130 |
|
131 |
adv_image = to_pil(adv_X[0]).convert("RGB")
|
132 |
adv_image = recover_image(adv_image, init_image, mask_image, background=True)
|
|
|
107 |
target_image = target_image.resize((512, 512))
|
108 |
|
109 |
with torch.autocast('cuda'):
|
110 |
+
mask, X = prepare_mask_and_masked_image(init_image, mask_image)
|
111 |
+
X = X.half().cuda()
|
112 |
+
mask = mask.half().cuda()
|
113 |
+
|
114 |
+
# Here we attack towards the embedding of a random target image. You can also simply attack towards an embedding of zeros!
|
115 |
+
target = pipe_inpaint.vae.encode(preprocess(target_image).half().cuda()).latent_dist.mean
|
116 |
|
117 |
+
adv_X = pgd(X,
|
118 |
+
target = target,
|
119 |
+
model=pipe_inpaint.vae.encode,
|
120 |
+
criterion=torch.nn.MSELoss(),
|
121 |
+
clamp_min=-1,
|
122 |
+
clamp_max=1,
|
123 |
+
eps=0.06,
|
124 |
+
step_size=0.01,
|
125 |
+
iters=1000,
|
126 |
+
mask=1-mask
|
127 |
+
)
|
128 |
+
|
129 |
+
adv_X = (adv_X / 2 + 0.5).clamp(0, 1)
|
|
|
|
|
|
|
130 |
|
131 |
adv_image = to_pil(adv_X[0]).convert("RGB")
|
132 |
adv_image = recover_image(adv_image, init_image, mask_image, background=True)
|