Spaces:
Sleeping
Sleeping
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("
|
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("
|
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
|