updates

ultralytics · Dec 12, 2018 · 5c92972 · 5c92972
1 parent e17c05d
commit 5c92972
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -42,7 +42,7 @@ def detect(opt):
 
     # Load model 1
     model = Darknet(opt.cfg, opt.img_size)
-    checkpoint = torch.load('weights/best.pt', map_location='cpu')
+    checkpoint = torch.load('weights/xview_best_lite.pt', map_location='cpu')
 
     model.load_state_dict(checkpoint['model'])
     model.to(device).eval()