Calculer l'IoU

On vous demande 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 vérité terrain bbox.

Laquelle des boîtes prédites est la plus proche de la vérité terrain ?

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