From 52ecf50607de7e199f6de36637053324555dc37d Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Sat, 9 May 2020 11:14:07 +0800 Subject: [PATCH] fix clip_bbox (#620) --- 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 7af4cd324..dfcec0d87 100644 --- a/ppdet/modeling/anchor_heads/yolo_head.py +++ b/ppdet/modeling/anchor_heads/yolo_head.py @@ -384,7 +384,7 @@ class YOLOv4Head(YOLOv3Head): yolo_loss=yolo_loss, iou_aware=iou_aware, iou_aware_factor=iou_aware_factor, - clip_box=clip_bbox) + clip_bbox=clip_bbox) self.spp_stage = spp_stage def _upsample(self, input, scale=2, name=None): -- GitLab