Learn R Programming

torchvision (version 0.7.0)

generalized_box_iou: Generalized Box IoU

Description

Return generalized intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in \((x_{min}, y_{min}, x_{max}, y_{max})\) format with \(0 \leq x_{min} < x_{max}\) and \(0 \leq y_{min} < y_{max}\).

Usage

generalized_box_iou(boxes1, boxes2)

Value

generalized_iou (Tensor[N, M]): the NxM matrix containing the pairwise generalized_IoU values for every element in boxes1 and boxes2

Arguments

boxes1

(Tensor[N, 4])

boxes2

(Tensor[M, 4])

Details

Implementation adapted from https://github.com/facebookresearch/detr/blob/master/util/box_ops.py