未验证 提交 3268ed1f 编写于 作者: G Guanghua Yu 提交者: GitHub

fix faceboxes steps in prior_box (#1255)

上级 a77455bc
...@@ -56,7 +56,7 @@ class FaceBoxes(object): ...@@ -56,7 +56,7 @@ class FaceBoxes(object):
densities=[[4, 2, 1], [1], [1]], densities=[[4, 2, 1], [1], [1]],
fixed_sizes=[[32., 64., 128.], [256.], [512.]], fixed_sizes=[[32., 64., 128.], [256.], [512.]],
num_classes=2, num_classes=2,
steps=[16., 32., 64.]): steps=[8., 16., 32.]):
super(FaceBoxes, self).__init__() super(FaceBoxes, self).__init__()
self.backbone = backbone self.backbone = backbone
self.num_classes = num_classes self.num_classes = num_classes
...@@ -117,7 +117,7 @@ class FaceBoxes(object): ...@@ -117,7 +117,7 @@ class FaceBoxes(object):
fixed_ratios=[1.], fixed_ratios=[1.],
clip=False, clip=False,
offset=0.5, offset=0.5,
steps=[self.steps[i]]) steps=[self.steps[i]] * 2)
num_boxes = box.shape[2] num_boxes = box.shape[2]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册