Calculer l'IoU

On vous a demandé de calculer la métrique Intersection over Union (IoU) entre chacune des trois boîtes englobantes prédites (box_a,box_b,box_c) et la boîte de référence bbox.

Quelle boîte prédite est la plus proche de la réalité ?

box_iou a été importé depuis torchvision.ops. box_a``box_b, box_c, bbox ont déjà été convertis en tenseurs de type torch.int avec une dimension batch supplémentaire.