未验证 提交 a43414a5 编写于 作者: J JiaQi Xu 提交者: GitHub

Update ssd.py

上级 9be4c9c5
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册