diff --git a/ssd.py b/ssd.py index b456149560f292657c69877c70e548eddb793e65..ec3eadd2fc9f3bda458ec19151da31b29733815e 100644 --- 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', - "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