From 6b88946f8d0e6ae64278ba447e425c7591374858 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Wed, 28 Oct 2020 11:22:19 +0800 Subject: [PATCH] update yolo_box (#1616) --- ppdet/modeling/layers.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ppdet/modeling/layers.py b/ppdet/modeling/layers.py index 671907898..424e9017d 100644 --- a/ppdet/modeling/layers.py +++ b/ppdet/modeling/layers.py @@ -401,9 +401,8 @@ class YOLOBox(object): self.clip_bbox = clip_bbox def __call__(self, x, img_size, anchors, num_classes, stage=0): - outs = fluid.layers.yolo_box(x, img_size, anchors, num_classes, - self.conf_thresh, self.downsample_ratio // - 2**stage, self.clip_bbox) + outs = ops.yolo_box(x, img_size, anchors, num_classes, self.conf_thresh, + self.downsample_ratio // 2**stage, self.clip_bbox) return outs -- GitLab