Calcola l'IoU

Ti è stato chiesto di calcolare la metrica Intersection over Union (IoU) tra ciascuna delle tre bounding box previste (box_a,box_b,box_c) e la bounding box di ground truth bbox.

Quale box prevista è più vicina alla ground truth?

box_iou è stata importata da torchvision.ops. box_a, box_b, box_c, bbox sono già state convertite in tensori di tipo torch.int con un’ulteriore dimensione batch.