我有图像numpy数组(640,480,3),其中X,Y是坐标,(255 255 255)是点的颜色掩码
我尝试获得新的2d数组(x,y),其中X和Y是当颜色>时点的坐标。零..
我试试这段代码 并且它可以工作,但它需要太多的处理器时间
<ng-content>
答案 0 :(得分:2)
您可以对最后一个轴的第一个元素进行切片,将其与0
进行比较,然后使用np.argwhere
获取这些索引,即x
,y
坐标在(N,2)
形状的阵列中。
因此,实施将简单地 -
new = np.argwhere( edges[...,0]!=0 )