Update ssd.py

ea63a17d · JiaQi Xu · GitHub · eedf9486 · ea63a17d
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ssd.py ssd.py +2 -2

未找到文件。
--- a/ssd.py
+++ b/ssd.py
@@ -14,8 +14,8 @@ from torch.autograd import Variable
 MEANS = (104, 117, 123)
 class SSD(object):
    _defaults = {
-        "model_path": 'logs/Epoch42-Loc0.5130-Conf1.4331',
+        "model_path": 'model_data/ssd_weights.pth',
-        "classes_path": 'model_data/new_classes.txt',
+        "classes_path": 'model_data/voc_classes.txt',
        "model_image_size" : (300, 300, 3),
        "confidence": 0.5,
        "Cuda": True