未验证 提交 50f32ee1 编写于 作者: L lxastro 提交者: GitHub

fix iou aware bug (#578)

上级 95c440d5
......@@ -117,6 +117,7 @@ class YOLOv3Loss(object):
for i, (output, target,
anchors) in enumerate(zip(outputs, targets, mask_anchors)):
an_num = len(anchors) // 2
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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册