Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-08-01 21:58:43 +02:00
parent 56f38ed6a2
commit 79111cb18f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
models.py
View File

@ -265,13 +265,15 @@ def load_darknet_weights(self, weights, cutoff=-1):
file = Path(weights).name file = Path(weights).name
# Try to download weights if not available locally # Try to download weights if not available locally
msg = weights + ' missing, download from https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI'
if not os.path.isfile(weights): if not os.path.isfile(weights):
try: try:
url = 'https://pjreddie.com/media/files/' + file url = 'https://pjreddie.com/media/files/' + file
print('Downloading ' + url) print('Downloading ' + url)
os.system('curl ' + url + ' -o ' + weights) os.system('curl -f ' + url + ' -o ' + weights)
except IOError: except IOError:
print(weights + ' not found.\nTry https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI') print(msg)
assert os.path.exists(weights), msg # download missing weights from Google Drive
# Establish cutoffs # Establish cutoffs
if file == 'darknet53.conv.74': if file == 'darknet53.conv.74':