返回索引检测到的同一对象

Question

我是工作对象检测，在检测到下面的multidimention数组的输出示例之后：

[[2433 1629 2656 1910]
 [2275  962 2538 1264]
 [2284 1191 2533 1472]
 [2303 1380 2551 1662]
 [2461 1442 2694 1732]
 [2284 1558 2533 1852]
 [2433 1261 2671 1524]
 [...................]
                      ]

有一个矩阵[nx4]，包括x，y，宽度，高度 现在我想返回具有相同类ID的对象的索引/数据。如何在Python或/和Numpy中实现？

Answer 1

假设您已将类存储在形状为classes的数组(n,)中，以便矩阵M的第一行与数组的第一个元素{{1}相关联}}。

鉴于任何课程classes，您可以使用

找到属于班级c的所有行

c

矩阵中的相应行由

给出

classes == c