wizzseen commited on
Commit
b567a22
·
verified ·
1 Parent(s): 5ca7066

Update TryYours-Virtual-Try-On/get_cloth_mask.py

Browse files
TryYours-Virtual-Try-On/get_cloth_mask.py CHANGED
@@ -11,7 +11,7 @@ warnings.filterwarnings("ignore")
11
 
12
  model = create_model("Unet_2020-10-30")
13
  model.eval()
14
- image = load_rgb("./static/cloth_web.jpg")
15
 
16
  -0
17
  transform = albu.Compose([albu.Normalize(p=1)], p=1)
@@ -30,7 +30,7 @@ mask = unpad(mask, pads)
30
  img = np.full((1024, 768, 3), 255)
31
  seg_img = np.full((1024, 768), 0)
32
 
33
- b = cv2.imread("./static/cloth_web.jpg")
34
  b_img = mask * 255
35
 
36
  # Calculate the exact dimensions that will fit into the img array
 
11
 
12
  model = create_model("Unet_2020-10-30")
13
  model.eval()
14
+ image = load_rgb("TryYours-Virtual-Try-On/static/cloth_web.jpg")
15
 
16
  -0
17
  transform = albu.Compose([albu.Normalize(p=1)], p=1)
 
30
  img = np.full((1024, 768, 3), 255)
31
  seg_img = np.full((1024, 768), 0)
32
 
33
+ b = cv2.imread("TryYours-Virtual-Try-On/static/cloth_web.jpg")
34
  b_img = mask * 255
35
 
36
  # Calculate the exact dimensions that will fit into the img array