在布尔张量掩码中查找峰值点的索引（第一个 True）

Question

在 pytorch 中运行一个 Detectron2 模型后，Detectron2 给我它找到的对象掩码作为（真/假）张量。在图像中发现了 33 个对象，所以我有 torch.Size([33, 683, 1024])。

tensor([[False, False, False,  ..., False, False, False],
    [False, False, False,  ..., False, False, False],
    [False, False, False,  ..., False, False, False],
    ...,
    [False, False, False,  ..., False, False, False],
    [False, False, False,  ..., False, False, False],
    [False, False, False,  ..., False, False, False]], device='cuda:0')

到目前为止，这很棒。但我需要这 33 个对象的 y 维度（高度）的峰值坐标。 （假设物体是气球，那么我需要气球的顶部作为（x，y）点）

知道如何尽快获得峰值点坐标 提前致谢

Answer 1

我已经遍历每个维度并检查是否满足 True 条件，但需要几分钟才能找到索引

然后我使用了torch.where方法，它立即找到了所有满足条件的索引。

for maskCounter in range(masks.shape[0]):
    print((torch.where(masks[maskCounter] == True)[0][0]).item(), (torch.where(masks[maskCounter] == True)[1][0]).item())