iou_similarity

    IOU Similarity Operator

    计算两个框列表的intersection-over-union(IOU)。框列表

    应为LoDTensor,

    iou_similarity - 图1

    是普通张量,

    成批输入的所有实例共享

    iou_similarity - 图2

    中的框。给定框A和框B,IOU的运算如下:

    参数:

    返回:维度为

    iou_similarity - 图3

    的LoDTensor,代表每一对iou分数,数据类型与

    相同

    代码示例