From dd913d01585a2076b37e5466f183541798719e81 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 20 Sep 2019 13:21:57 +0200 Subject: [PATCH] updates --- models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models.py b/models.py index cf3feeb2..1ca41cf6 100755 --- a/models.py +++ b/models.py @@ -405,7 +405,7 @@ def attempt_download(weights): # Attempt to download pretrained weights if not found locally msg = weights + ' missing, download from https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI' - if not os.path.isfile(weights): + if weights and not os.path.isfile(weights): file = Path(weights).name if file == 'yolov3-spp.weights': @@ -430,4 +430,4 @@ def attempt_download(weights): print(msg) os.system('rm ' + weights) # remove partial downloads - assert os.path.exists(weights), msg # download missing weights from Google Drive + assert os.path.exists(weights), msg # download missing weights from Google Drive