From 6aed6c5fd0b31b1053d538095d4488eff8d5ad88 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 21 Apr 2020 09:13:38 -0700 Subject: [PATCH] attempt_download() update for '' weights --- models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models.py b/models.py index 84b6af88..9d0d7de5 100755 --- a/models.py +++ b/models.py @@ -441,9 +441,10 @@ def convert(cfg='cfg/yolov3-spp.cfg', weights='weights/yolov3-spp.weights'): def attempt_download(weights): # Attempt to download pretrained weights if not found locally + weights = weights.strip() msg = weights + ' missing, try downloading from https://drive.google.com/open?id=1LezFG5g3BCW6iYaV89B2i64cqEUZD7e0' - if weights and not os.path.isfile(weights): + if len(weights) > 0 and not os.path.isfile(weights): d = {'yolov3-spp.weights': '16lYS4bcIdM2HdmyJBVDOvt3Trx6N3W2R', 'yolov3.weights': '1uTlyDWlnaqXcsKOktP5aH_zRDbfcDp-y', 'yolov3-tiny.weights': '1CCF-iNIIkYesIDzaPvdwlcf7H9zSsKZQ',