Calcular IoU

Você precisa calcular a métrica Intersection over Union (IoU) entre cada uma das três caixas delimitadoras previstas (box_a,box_b,box_c) e a caixa de referência bbox.

Qual caixa prevista tá mais perto da verdade?

box_iou foi importado de torchvision.ops. box_a, box_b, box_c, bbox já foram convertidos para tensores do tipo torch.int com uma dimensão de lote adicional.