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

updates

This commit is contained in:
Glenn Jocher 2019-09-20 13:21:57 +02:00
parent 0e52b8f361
commit dd913d0158
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models.py
View File

@ -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