Spaces:
Running
Running
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
|