update yolo_box (#1616)

6b88946f · wangguanzhong · GitHub · 5bee2353 · 6b88946f
显示空白变更内容
内联并排

Showing with 2 addition and 3 deletion

ppdet/modeling/layers.py ppdet/modeling/layers.py +2 -3

未找到文件。
--- 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,
+        outs = ops.yolo_box(x, img_size, anchors, num_classes, self.conf_thresh,
-                                     self.conf_thresh, self.downsample_ratio //
+                            self.downsample_ratio // 2**stage, self.clip_bbox)
-                                     2**stage, self.clip_bbox)
        return outs