Spaces:
Sleeping
Sleeping
Update warp_design_on_dress.py
Browse files- warp_design_on_dress.py +1 -1
warp_design_on_dress.py
CHANGED
@@ -64,7 +64,7 @@ def run_design_warp_on_dress(dress_path, design_path, gmm_ckpt, tom_ckpt, output
|
|
64 |
gray = dress_tensor.mean(dim=1, keepdim=True)
|
65 |
|
66 |
# Create edge detection kernel
|
67 |
-
kernel = torch.tensor(
|
68 |
[[[-1, -1, -1],
|
69 |
[-1, 8, -1],
|
70 |
[-1, -1, -1]]], dtype=torch.float32, device=device)
|
|
|
64 |
gray = dress_tensor.mean(dim=1, keepdim=True)
|
65 |
|
66 |
# Create edge detection kernel
|
67 |
+
kernel = torch.tensor(
|
68 |
[[[-1, -1, -1],
|
69 |
[-1, 8, -1],
|
70 |
[-1, -1, -1]]], dtype=torch.float32, device=device)
|