如何获得布尔numpy掩码的左下角和右下角

Question

我有一个形状为(1024,1024) 的布尔掩码numpy 数组，它是图片上一个人的布尔掩码，如下所示。现在我想获取掩码中最左下角的点（标记为红色）、掩码中最右下角的点（标记为蓝色）的 X 和 Y 坐标（数组中的索引）。更具挑战性的是，我想得到下图中绿色和粉色点的索引（图像上人的肩膀顶部）。

有没有简单的方法可以得到这些点的坐标？

获取完全相同格式的掩码的示例代码如下：

from skimage import io

image = io.imread('https://i.imgur.com/X9BKDtl.png')
image = image.astype(bool)
image = image[:, :, 0]

Answer 1

我的幼稚方法是找到和 x + y 的最大值，其中 x 和 y 是图像上等于 True 的坐标。要获取左下角的索引，我们使用x - y

可以通过使用numpy.where 来获取索引来实现这一点：

import numpy as np
from skimage import io
import matplotlib.pyplot as plt


image = io.imread('https://i.imgur.com/X9BKDtl.png')
image = image.astype(bool)
image = image[:, :, 0]

x, y = np.where(image)  # get the indices (x, y) where the image is True

# Create a dictionary where the keys equate to the sum x + y
# or x - y respectively and the values are the corresponding
# x and y coordinates.

map_left = {x_i - y_i: (x_i, y_i) for x_i, y_i in zip(x, y)}
max_left = max(map_left.keys())
x_left, y_left = map_left[max_left]

map_right = {(x_i) + (y_i): (x_i, y_i) for x_i, y_i in zip(x, y)}
max_right = max(map_right.keys())
x_right, y_right = map_right[max_right]

plt.imshow(image)
# Note: x and y are swapped here due to the way the
# image is plotted
plt.plot(y_left, x_left, "x")
plt.plot(y_right, x_right, "x")

使用此代码，我得到以下结果