From 0f3f6c03e77ea340a6f992ac9cc756bf891f16a7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 19 Sep 2019 18:09:16 +0200 Subject: [PATCH] updates --- detect.py | 1 + 1 file changed, 1 insertion(+) diff --git a/detect.py b/detect.py index 859a9202..846eb004 100644 --- a/detect.py +++ b/detect.py @@ -22,6 +22,7 @@ def detect(save_txt=False, save_img=False, view_img=False): model = Darknet(opt.cfg, img_size) # Load weights + attempt_download(weights) if weights.endswith('.pt'): # pytorch format model.load_state_dict(torch.load(weights, map_location=device)['model']) else: # darknet format