Pytorch torch.stack使用

生成锚框
w=tensor([0.75, 0.50, 0.25, 1.06, 0.53])
h=tensor([0.75, 0.50, 0.25, 0.53, 1.06])
torch.stack((-w, -h, w, h))

tensor([[-0.75, -0.50, -0.25, -1.06, -0.53],
        [-0.75, -0.50, -0.25, -0.53, -1.06],
        [ 0.75,  0.50,  0.25,  1.06,  0.53],
        [ 0.75,  0.50,  0.25,  0.53,  1.06]])

w代表每个框的宽 h代表高

每一列代表一个框的坐标再除以二 就可以根据每个框的中间位置算出左上角和右下角的坐标