Calcular IoU

Te piden calcular la métrica Intersection over Union (IoU) entre cada una de las tres cajas delimitadoras predichas (box_a,box_b,box_c) y la caja de la verdad terreno bbox.

¿Qué caja predicha está más cerca de la verdad terreno?

box_iou se ha importado desde torchvision.ops. box_a, box_b, box_c, bbox ya se han convertido en tensores de tipo torch.int con una dimensión adicional de lote.