关于分析图像阵列的唯一值

Question

有以下功能可以读取图像，我添加了几行来重新输出图像，并输出图像数组的不同像素值。图像看起来像这样。然而      a，indices = np.unique（img，return_index = True）

给出

a [0 1]

指数[0 879385]

似乎图像数组有两个唯一值[0 1]，这是有意义的，但索引表示什么？

def _get_image_data_pil(image_id, image_type, return_exif_md=False, return_shape_only=False):
    fname = get_filename(image_id, image_type)
    try:
        img_pil = Image.open(fname)
    except Exception as e:
        assert False, "Failed to read image : %s, %s. Error message: %s" % (image_id, image_type, e)

    if return_shape_only:
        return img_pil.size[::-1] + (len(img_pil.getbands()),)
    # -----
    # this is what I adde
    # -----

    img = np.asarray(img_pil)
    plt.imshow(img)
    plt.show()
    a,indices =np.unique(img,return_index=True)
    print('a ',a)
    print('indices ',indices)

    assert isinstance(img, np.ndarray), "Open image is not an ndarray. Image id/type : %s, %s" % (image_id, image_type)
    if not return_exif_md:
        return img
    else:
        return img, img_pil._getexif()

Answer 1

索引在 flattened 输入数组中首次出现唯一值。要将这些索引转换为2-d索引到输入中，您可以使用np.unravel_index。

例如，假设img的形状为（1000,1600）：

In [110]: shape = (1000, 1600)

In [111]: indices = [0, 879385]

In [112]: np.unravel_index(indices, shape)
Out[112]: (array([  0, 549]), array([  0, 985]))

np.unravel_indices返回了两个数组，每个维度一个。也就是说，第一阵列保持第一维的索引（即行），第二阵列保持第二维的索引（即列）。要将这些放入坐标数组中，您可以使用np.column_stack：

In [113]: np.column_stack(np.unravel_index(indices, shape))
Out[113]: 
array([[  0,   0],
       [549, 985]])