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-19 17:19:43 +02:00
parent dc445f42bf
commit 6fa58d3c40
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -301,6 +301,8 @@ def load_darknet_weights(self, weights, cutoff=-1):
gdrive_download(id='1oPCHKsM2JpM-zgyepQciGli9X0MTsJCO', name=weights)
elif file == 'darknet53.conv.74':
gdrive_download(id='18xqvs_uwAqfTXp-LJCYLYNHBOcrwbrp0', name=weights)
elif file == 'yolov3-spp.pt':
gdrive_download(id='1vFlbJ_dXPvtwaLLOu-twnjK4exdFiQ73', name=weights)
else:
try: # download from pjreddie.com
url = 'https://pjreddie.com/media/files/' + file