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

TryYours-Virtual-Try-On/main.py

@@ -13,14 +13,15 @@ if __name__ == '__main__':
     opt = parser.parse_args()
     
     # Read input image
-    img=cv2.imread("./static/origin_web.jpg")
+
+    img=cv2.imread("TryYours-Virtual-Try-On/static/origin_web.jpg")
     ori_img=cv2.resize(img,(768,1024))
-    cv2.imwrite("./origin.jpg",ori_img)
+    cv2.imwrite("TryYours-Virtual-Try-On/static/origin.jpg",ori_img)
 
     # Resize input image
-    img=cv2.imread('origin.jpg')
+    img=cv2.imread('TryYours-Virtual-Try-On/static/origin.jpg')
     img=cv2.resize(img,(384,512))
-    cv2.imwrite('resized_img.jpg',img)
+    cv2.imwrite('TryYours-Virtual-Try-On/static/resized_img.jpg',img)
 
     # Get mask of cloth
     print("Get mask of cloth\n")
@@ -34,7 +35,7 @@ if __name__ == '__main__':
 
     # Generate semantic segmentation using Graphonomy-Master library
     print("Generate semantic segmentation using Graphonomy-Master library\n")
-    os.chdir("./Graphonomy-master")
+    os.chdir("TryYours-Virtual-Try-On/static/Graphonomy-master")
     terminnal_command ="python exp/inference/inference.py --loadmodel ./inference.pth --img_path ../resized_img.jpg --output_path ../ --output_name /resized_segmentation_img"
     os.system(terminnal_command)
     os.chdir("../")