gaur3009 commited on
Commit
1964060
·
verified ·
1 Parent(s): 40d3928

Update warp_design_on_dress.py

Browse files
Files changed (1) hide show
  1. 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)