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

Update ssd.py

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