IoU की गणना करें

आपसे तीन प्रेडिक्टेड बाउंडिंग बॉक्स (box_a,box_b,box_c) और ग्राउंड ट्रुथ बॉक्स bbox के बीच Intersection over Union (IoU) मैट्रिक निकालने के लिए कहा गया है.

कौन-सा प्रेडिक्टेड बॉक्स ग्राउंड ट्रुथ के सबसे करीब है?

box_iou को torchvision.ops से इम्पोर्ट किया गया है. box_a, box_b, box_c, bbox पहले से ही अतिरिक्त बैच डायमेंशन के साथ torch.int टाइप टेन्सर में कन्वर्ट किए गए हैं.