Update TryYours-Virtual-Try-On/posenet/converter/tfjs2pytorch.py

TryYours-Virtual-Try-On/posenet/converter/tfjs2pytorch.py

@@ -96,9 +96,9 @@ def convert(model_id, model_dir, output_stride=16, image_size=513, check=True):
     checkpoint_path = os.path.join(model_dir, checkpoint_name) + '.pth'
     torch.save(m.state_dict(), checkpoint_path)
 
-    if check and os.path.exists("./images/tennis_in_crowd.jpg"):
+    if check and os.path.exists("TryYours-Virtual-Try-On/posenet/converter/images/tennis_in_crowd.jpg"):
         # Result
-        input_image = _read_imgfile("./images/tennis_in_crowd.jpg", width, height)
+        input_image = _read_imgfile("TryYours-Virtual-Try-On/posenet/converter/images/tennis_in_crowd.jpg", width, height)
         input_image = np.array(input_image, dtype=np.float32)
         input_image = input_image.reshape(1, 3, height, width)
         input_image = torch.Tensor(input_image)