python - 如何从 PyTorch 张量中分割多个边界框？

Question

所以我的问题很简单：

• 我有多个由以下组成的边界框：[TopLeft_X, TopLeft_Y, BottomRight_X, BottomRight_Y]
• 而且我有一个张量面具大小[900, 1600]

• 所以这是我拥有的前 3 个边界框：

  tensor([[1041, 465, 1102, 508], [1037, 468, 1108, 519], [1041, 465, 1107, 515]], device='cuda:0', dtype=torch.int32)

• 本质上，我想获得每个边界框与蒙版相交的像素数（值为 1）。

• 目前，我正在遍历所有边界框并获得相应的切片：

  map_mask[y_min:y_max, x_min:x_max] -> 2D 输出

在 PyTorch 中没有 for 循环的情况下如何同时执行此操作？

标签： pythonpython-3.xpytorchslicebounding-box