From 58940e9793887c1b4bde01b74c1759865d59480c Mon Sep 17 00:00:00 2001 From: longxiang Date: Sun, 12 Jul 2020 10:52:36 +0800 Subject: [PATCH] fix coordconv bug --- ppdet/modeling/anchor_heads/yolo_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/anchor_heads/yolo_head.py b/ppdet/modeling/anchor_heads/yolo_head.py index 77b26b137..bb4beeb67 100644 --- a/ppdet/modeling/anchor_heads/yolo_head.py +++ b/ppdet/modeling/anchor_heads/yolo_head.py @@ -121,7 +121,7 @@ class YOLOv3Head(object): act='leaky', is_test=True, name=None): - if coord_conv: + if coord_conv and self.coord_conv: input = self._add_coord(input) conv = fluid.layers.conv2d( input=input, -- GitLab