Calculer l'IoU

On vous demande de calculer la mesure Intersection over Union (IoU) entre chacune des trois boîtes englobantes prédites (box_a, box_b, box_c) et la boîte réelle bbox.

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

box_iou a été importée de 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.