我有一个python图像处理函数,它试图获取图像的主颜色。我使用了我在这里找到的一个函数https://github.com/tarikd/python-kmeans-dominant-colors/blob/master/utils.py
它可以工作,但不幸的是,我不太明白它是做什么的,我了解到np.histogram
相当慢,我应该使用cv2.calcHist
,因为它的40倍快,根据这个:https://docs.opencv.org/trunk/d1/db7/tutorial_py_histogram_begins.html
我想知道如何更新代码以使用cv2.calcHist
或更好的值,我必须输入哪些值。
我的功能
def centroid_histogram(clt):
# grab the number of different clusters and create a histogram
# based on the number of pixels assigned to each cluster
num_labels = np.arange(0, len(np.unique(clt.labels_)) + 1)
(hist, _) = np.histogram(clt.labels_, bins=num_labels)
# normalize the histogram, such that it sums to one
hist = hist.astype("float")
hist /= hist.sum()
# return the histogram
return hist
字符串clt
的pprint
是这样的,不确定这是否有帮助
KMeans(algorithm='auto', copy_x=True, init='k-means++', max_iter=300,
n_clusters=1, n_init=10, n_jobs=1, precompute_distances='auto',
random_state=None, tol=0.0001, verbose=0)
型
我的代码可以在这里找到:https://github.com/primus852/python-movie-barcode
我是一个初学者,所以任何帮助都非常感谢。
根据要求:
示例镜像
的数据
最主色:
rgb(22,28,37)
个
直方图计算时间:
0.021515369415283203s
个
4条答案
按热度按时间nom7f22z1#
可以建议使用
np.unique
和np.bincount
来获得最主要的颜色的两种方法。此外,在链接页面中,它谈到bincount
作为更快的替代方案,所以这可能是要走的路。方法#1
字符串
方法#2
型
在
[0,9)
密集范围内对1000 x 1000
彩色图像进行验证和计时,以获得可重现的结果-型
进一步提高
利用
multi-core
和numexpr
模块进一步提升大数据处理能力-型
时间-
型
ulmd4ohb2#
@Divakar给出了一个很好的答案。但是如果你想将自己的代码移植到OpenCV,那么:
字符串
您的图像的结果:
主色为:bgr([41 31 23])
时间:0.10798478126525879秒
qlzsbp2j3#
cv2.calcHist()
的等效代码是替换:字符串
与
型
注意
cv2.calcHist
只接受uint8
和float32
作为元素类型。更新
看起来opencv和numpy的分箱方式是不同的,因为如果分箱的数量没有Map到值的范围,直方图就会不同:
型
apeeds0o4#
@Divakar的答案的改进版本。RGBA图像的单独解决方案。RGBA图像的完全透明像素不会被计算在内。
字符串