From 50f32ee1f460c115b8da5b6bcb02af37ccbe92b0 Mon Sep 17 00:00:00 2001 From: lxastro Date: Thu, 30 Apr 2020 13:06:21 +0800 Subject: [PATCH] fix iou aware bug (#578) --- ppdet/modeling/losses/yolo_loss.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ppdet/modeling/losses/yolo_loss.py b/ppdet/modeling/losses/yolo_loss.py index 6777608f7..92f00c5a5 100644 --- a/ppdet/modeling/losses/yolo_loss.py +++ b/ppdet/modeling/losses/yolo_loss.py @@ -117,7 +117,8 @@ class YOLOv3Loss(object): for i, (output, target, anchors) in enumerate(zip(outputs, targets, mask_anchors)): an_num = len(anchors) // 2 - ioup, output = self._split_ioup(output, an_num, num_classes) + if self._iou_aware_loss is not None: + ioup, output = self._split_ioup(output, an_num, num_classes) x, y, w, h, obj, cls = self._split_output(output, an_num, num_classes) tx, ty, tw, th, tscale, tobj, tcls = self._split_target(target) -- GitLab