Spaces:
Sleeping
Sleeping
Update TryYours-Virtual-Try-On/get_cloth_mask.py
Browse files
TryYours-Virtual-Try-On/get_cloth_mask.py
CHANGED
@@ -48,5 +48,5 @@ img[int((1024-exact_height)/2): 1024-int((1024-exact_height)/2), int((768-exact_
|
|
48 |
seg_img[int((1024-exact_height)/2): 1024-int((1024-exact_height)/2), int((768-exact_width)/2):768-int((768-exact_width)/2)] = b_img
|
49 |
|
50 |
# Save the images
|
51 |
-
cv2.imwrite("
|
52 |
-
cv2.imwrite("
|
|
|
48 |
seg_img[int((1024-exact_height)/2): 1024-int((1024-exact_height)/2), int((768-exact_width)/2):768-int((768-exact_width)/2)] = b_img
|
49 |
|
50 |
# Save the images
|
51 |
+
cv2.imwrite("TryYours-Virtual-Try-On/HR-VITON-main/test/test/cloth/00001_00.jpg", img)
|
52 |
+
cv2.imwrite("TryYours-Virtual-Try-On/HR-VITON-main/test/test/cloth-mask/00001_00.jpg", seg_img)
|