在pytorch中给出任意两个对角线坐标，在一批张量中获得靠近原点的矩形角

作者: Minions
发布时间: 2024-06-07 07:11:37 (1月前)
转自：

2 条回复

0#
回复此人
誓言你好 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 您可以迭代所有张量，并为每个张量计算到四个角的距离，并以最小距离拍摄角落。 </p> <pre> <code> import torch a = torch.tensor([[3,2,2,3], [1,1,2,2]]) c = torch.zeros(a.shape[0], 2) for idx, x in enumerate(a): d1 = x[0] ** 2 + x[1] ** 2 d2 = x[2] ** 2 + x[3] ** 2 d3 = x[0] ** 2 + x[3] ** 2 d4 = x[2] ** 2 + x[1] ** 2 dmin = min(d1, d2, d3, d4) if d1 == dmin: c[idx] = torch.tensor([x[0], x[1]]) elif d2 == dmin: c[idx] = torch.tensor([x[2], x[3]]) elif d3 == dmin: c[idx] = torch.tensor([x[0], x[3]]) elif d4 == dmin: c[idx] = torch.tensor([x[2], x[1]]) print(c) # tensor([[2., 2.], [1., 1.]]) </code> </pre> </DIV>

编辑

登录后才能参与评论