From a43414a5689b54d72013e42f2dc268772163a492 Mon Sep 17 00:00:00 2001
From: JiaQi Xu <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 2 Apr 2020 00:56:38 +0800
Subject: [PATCH] Update ssd.py

---
 ssd.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ssd.py b/ssd.py
index 5ab7ddb..1240c2b 100644
--- a/ssd.py
+++ b/ssd.py
@@ -14,7 +14,7 @@ from torch.autograd import Variable
 MEANS = (104, 117, 123)
 class SSD(object):
     _defaults = {
-        "model_path": 'logs/Epoch4-Loc0.5106-Conf1.3239.pth',
+        "model_path": 'model_data/ssd_weights.pth',
         "classes_path": 'model_data/new_classes.txt',
         "model_image_size" : (300, 300, 3),
         "confidence": 0.5,
-- 
GitLab