diff --git a/ppdet/modeling/anchor_heads/yolo_head.py b/ppdet/modeling/anchor_heads/yolo_head.py index 77b26b137e00bf0e9f64be721b7868a49040a0a7..bb4beeb677a01e12510f040bf939b2729a8970f8 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,