From d99a2efbc5eb896819971f8a19f73bdeafb007e3 Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Wed, 19 Aug 2020 17:31:57 +0800 Subject: [PATCH] fix faceboxes steps in prior_box (#1256) --- ppdet/modeling/architectures/faceboxes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppdet/modeling/architectures/faceboxes.py b/ppdet/modeling/architectures/faceboxes.py index 6606173cc..2d8abe6b9 100644 --- a/ppdet/modeling/architectures/faceboxes.py +++ b/ppdet/modeling/architectures/faceboxes.py @@ -56,7 +56,7 @@ class FaceBoxes(object): densities=[[4, 2, 1], [1], [1]], fixed_sizes=[[32., 64., 128.], [256.], [512.]], num_classes=2, - steps=[16., 32., 64.]): + steps=[8., 16., 32.]): super(FaceBoxes, self).__init__() self.backbone = backbone self.num_classes = num_classes @@ -117,7 +117,7 @@ class FaceBoxes(object): fixed_ratios=[1.], clip=False, offset=0.5, - steps=[self.steps[i]]) + steps=[self.steps[i]] * 2) num_boxes = box.shape[2] -- GitLab