未验证 提交 c1c537b7 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix diou_loss_yolo.py scale_x_y & eps not set (#1239)

上级 c9c28040
......@@ -65,9 +65,10 @@ class DiouLossYolo(IouLoss):
eps (float): the decimal to prevent the denominator eqaul zero
'''
x1, y1, x2, y2 = self._bbox_transform(
x, y, w, h, anchors, downsample_ratio, batch_size, False)
x1g, y1g, x2g, y2g = self._bbox_transform(
tx, ty, tw, th, anchors, downsample_ratio, batch_size, True)
x, y, w, h, anchors, downsample_ratio, batch_size, False, 1.0, eps)
x1g, y1g, x2g, y2g = self._bbox_transform(tx, ty, tw, th, anchors,
downsample_ratio, batch_size,
True, 1.0, eps)
#central coordinates
cx = (x1 + x2) / 2
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册