PyTorch 自带的IoU计算，只需提供两个检测框box1， box2的对角位置[X1, Y1, X2, Y2], [X3, Y3, X4, Y4]即可:

• 记box1：左下坐标为[X1, Y1], 右上坐标为 [X2, Y2]
• 记box1：左下坐标为[X3, Y3], 右上坐标为 [X4, Y4]

$$J(A,B)=\frac{|A\cap B|}{|A|\cup |B|}=\frac{|A|\cap |B|}{|A|+|B|-|A\cap B|}$$

import torch
import torchvision.ops.boxes as bops

# box1 = torch.tensor([[511, 41, 577, 76]], dtype=torch.float)
box1 = torch.tensor([[544, 59, 610, 94]], dtype=torch.float)
box2 = torch.tensor([[544, 59, 610, 94]], dtype=torch.float)
iou = bops.box_iou(box1, box2).numpy()[0,0]
print(iou)
# tensor([[0.1382]])