Spaces:
Runtime error
Runtime error
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)
|