使用opencv和python的图像中所有颜色的列表

Question

Python初学者在这里.我正在使用python 3.6和opencv,我正在尝试创建一个图像中存在的所有颜色的rgb值列表.我可以使用cv2.imread后跟image [px,px]读取一个像素的rgb值.结果是一个numpy数组.我想要的是图像中存在的独特颜色的rgb元组列表,我不知道该怎么做.任何帮助表示赞赏.TIA

Answer 1

看看numpy的numpy.unique()函数:

import numpy as np

test = np.array([[255,233,200], [23,66,122], [0,0,123], [233,200,255], [23,66,122]])
print(np.unique(test, axis=0, return_counts = True))

>>> (array([[  0,   0, 123],
       [ 23,  66, 122],
       [233, 200, 255],
       [255, 233, 200]]), array([1, 2, 1, 1]))

您可以在2D numpy数组中收集RGB numpy数组,然后使用带axis=0参数的numpy.unique()函数来遍历其中的数组.可选参数return_counts=True还将为您提供出现的次数.