updates
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
8f913ba82a
commit
32b54c81d5
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue