Update utils.py

utils.py

@@ -1,3 +1,4 @@
+import os
 import cv2
 import torch
 from urllib.request import urlretrieve
@@ -26,6 +27,9 @@ def accuracy(predictions, ground_truth):
     return score.item()
 
 def download_weights(url):
+    cd = os.getcwd()
     fname = url.split('/')[-1]
-    urlretrieve(url, fname)
+    fname = os.path.join(cd, fname)
+    if not os.path.exists(fname):
+        urlretrieve(url, fname)
     return fname