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 타입의 텐서로 변환되어 있습니다.