如何在一批图像中找到具有特定值的图像的位置答案

【问题标题】：How to find locations of images with a particular value in a batch of images如何在一批图像中找到具有特定值的图像的位置
【发布时间】：2019-07-06 02:02:46
【问题描述】：

我有一批形状为(N, C, H, W) 的图像，其中 N 是图像数量，C - 通道数，H,W - 高度和宽度。

每个图像有 2 个通道，其中一些像素值为 [-1 , -1]。

如何在不使用 for 循环的情况下找到批处理中这些像素的位置，因为它非常慢。

【问题讨论】：

标签： python numpy

【解决方案1】：

使用numpy.where:

# creating test data
test = np.zeros((5,2,3,3))
test[3,:,2,1] = [-1.,-1.]

value = -np.ones((1.,2.,1.,1.)) # this is the value you are looking for
np.where(test == value)

# this returns: (array([3, 3], dtype=int64),
#                array([0, 1], dtype=int64),
#                array([2, 2], dtype=int64),
#                array([1, 1], dtype=int64))

编辑：要获得相应的掩码，请不要使用where：

test == value

【讨论】：

是否可以获得具有批次形状但在各自位置具有True 或False 的掩码数组。
是的，我扩展了我的答案

【解决方案2】：

您可以使用numpy.where。一个简单的例子：

x = np.random.randn(4,2,10,10)
x[0,1,2,3] = 1
x[0,1,4,5] = 1
np.where(x==1)

(数组([0, 0], dtype=int64), 数组([1, 1], dtype=int64), 数组([2, 4], dtype=int64), 数组([3, 5], dtype=int64))

【讨论】：