如何计算图像中的 RGB 或 HSV 通道组合？

Question

我使用 python opencv 加载一个形状为 (30, 100, 3) 的图像，现在想计算所有颜色的频率，按颜色，我不是指单个通道，我指的是通道组合。含义 3 个通道列表，例如 [255, 0, 0] 代表红色，[255, 255, 0] 代表黄色，[100, 100, 100] 代表另一种颜色。所以我希望将最后一个轴（通道）视为一个整体并计算其频率。

opencv 或 numpy 中是否有任何内置函数可以轻松地将 3 通道列表视为一个元素并计算其频率？

Answer 1

您可以使用np.unique它的新axis参数功能进行分组 -

np.c_[np.unique(im.reshape(-1,3), axis=0, return_counts=1)]

样品运行 -

In [56]: im
Out[56]: 
array([[[255, 255, 255],
        [255,   0,   0]],

       [[255,   0, 255],
        [255, 255, 255]]])

In [57]: np.c_[np.unique(im.reshape(-1,3), axis=0, return_counts=1)]
Out[57]: 
array([[255,   0,   0,   1],
       [255,   0, 255,   1],
       [255, 255, 255,   2]])