Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2019-09-02 12:09:00 +02:00
parent 8f913ba82a
commit 32b54c81d5
1 changed files with 1 additions and 0 deletions

View File

@ -302,6 +302,7 @@ def load_darknet_weights(self, weights, cutoff=-1):
os.system('curl -f ' + url + ' -o ' + weights)
except IOError:
print(msg)
os.system('rm ' + weights) # remove partial downloads
assert os.path.exists(weights), msg # download missing weights from Google Drive
# Establish cutoffs