我是工作对象检测,在检测到多维数组的输出示例如下:
[[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,width,height,现在我想返回具有相同类id的对象的索引/数据。如何在Python或/和Numpy中实现?
发布于 2018-05-22 20:24:23
假设您已经将类存储在一个形状为(n,)的数组classes中,因此矩阵M的第一行与数组classes的第一个元素相关联。
给定任何类c,都可以使用以下命令查找属于类c的所有行
classes == c矩阵中的相应行由下式给出
M[classes == c]https://stackoverflow.com/questions/50464623
复制相似问题