From 6fa58d3c400a527984f04e0ebaae19d44512f0f1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 19 Sep 2019 17:19:43 +0200 Subject: [PATCH] updates --- models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models.py b/models.py index 054f2d09..e866ad6b 100755 --- a/models.py +++ b/models.py @@ -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