diff --git a/ssd.py b/ssd.py index 5ab7ddb8c6c1bbe4f815d0f84f21488b175c7afb..1240c2b1c23ba1c295e65baab8dac650eae0a711 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,