Pycocotools best.pt after COCO train (#1593)

2020-12-06 14:58:50 +01:00
parent adc49abc71
commit 4a07280884
3 changed files with 25 additions and 15 deletions
@@ -17,7 +17,7 @@ def gsutil_getsize(url=''):

 def attempt_download(weights):
    # Attempt to download pretrained weights if not found locally
-    weights = weights.strip().replace("'", '')
+    weights = str(weights).strip().replace("'", '')
    file = Path(weights).name.lower()

    msg = weights + ' missing, try downloading from https://github.com/ultralytics/yolov3/releases/'