Update ssd.py

JoahHe · Mar 20, 2020 · ea63a17 · ea63a17
1 parent eedf948
commit ea63a17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ssd.py b/ssd.py
@@ -14,8 +14,8 @@
 MEANS = (104, 117, 123)
 class SSD(object):
     _defaults = {
-        "model_path": 'logs/Epoch42-Loc0.5130-Conf1.4331',
-        "classes_path": 'model_data/new_classes.txt',
+        "model_path": 'model_data/ssd_weights.pth',
+        "classes_path": 'model_data/voc_classes.txt',
         "model_image_size" : (300, 300, 3),
         "confidence": 0.5,
         "Cuda": True