2

我正在尝试了解 CNN 中边界框回归的损失函数。目前我使用 Lasagne 和 Theano，这使得编写损失表达式非常容易。许多消息来源提出了不同的方法，我问自己在实践中通常使用哪一种。

边界框坐标按顺序表示为归一化坐标[left, top, right, bottom]（使用T.matrix('targets', dtype=theano.config.floatX)）。

到目前为止，我已经尝试了以下功能；但是，它们都有其缺点。

联合交叉口

有人建议我使用该Intersection over Union度量来确定 2 个边界框对齐和重叠的程度。但是，当框不重叠并且交集为0时会出现问题；然后无论边界框相距多远，整个商都变为 0。我将其实现为：

def get_area(A):
    return (A[:,2] - A[:,0]) * (A[:,1] - A[:,3])

def get_intersection(A, B):
    return (T.minimum(A[:,2], B[:,2]) - T.maximum(A[:,0], B[:,0])) \
        * (T.minimum(A[:,1], B[:,1]) - T.maximum(A[:,3], B[:,3]))

def bbox_overlap_loss(A, B):
    """Computes the bounding box overlap using the
    Intersection over union"""
    intersection = get_intersection(A, B)
    union = get_area(A) + get_area(B) - intersection
    # Turn into loss
    l = 1.0 - intersection / union
    return l.mean()

平方直径差

为了为非重叠边界框创建误差度量，我尝试计算边界框直径的平方差。它似乎有效，但我几乎可以肯定有更好的方法来做到这一点。我将其实现为：

def squared_diameter_loss(A, B):
    # Represent the squared distance from the real diameter
    # in normalized pixel coordinates
    l = (abs(A[:,0:2]-B[:,0:2]) + abs(A[:,2:4]-B[:,2:4]))**2
    return l.mean()

欧几里得损失

最简单的函数是Euclidean Loss计算边界框参数平方差的平方根。但是，这并没有考虑重叠边界框的面积，而只考虑了参数left，right，top，bottom的差异。我将其实现为：

def euclidean_loss(A, B):
    l = lasagne.objectives.squared_error(A, B)
    return l.mean()

如果我在这里做错了什么，有人可以指导我对于这个用例或现场的边界框回归，哪个是最好的损失函数。实践中通常使用哪种损失函数？

4

1 回答 1

2

从个人的实施经验来看，与欧几里得（MSE 或 L2）损失相比，我使用 IOU 作为损失函数训练 CNN 的结果要好得多。没有使用平方直径差损失。一般来说，明确表示输出对于您希望完成的任务的优劣的损失函数可能是最好的。

对于值为 0 的 IOU，您可以在公式中引入一些附加项，使其优雅地趋向于 0，可能基于 bbox 中心之间的标准化距离。这可能会产生额外的效果，帮助将边界框相对于地面实况居中。

此响应主要是概念性的，但如果需要，我很乐意提供代码示例。

于 2019-09-04T13:59:47.157 回答