我有一个由 7 种不同可能颜色组成的 RGB 图像。我想以有效的方式计算图像中存在的每种像素类型的数量。所以如果可能的话,不要在每个像素上循环,至少不是手动(numpy 操作是可以的,因为它更快)我尝试将它加载到一个 numpy 数组中,这给了我一个 N*M*3 数组,但我想不出一种方法来计算特定值的像素......有什么想法吗?
2 回答
三国纷争
TA贡献1804条经验 获得超7个赞
只是部分展平并np.unique与return_counts = True和一起使用axis = 0
flat_image = image.reshape(-1, 3) # makes one long line of pixels
colors, counts = np.unique(flat_image, return_counts = True, axis = 0)
或者作为一行:
colors, counts = np.unique(image.reshape(-1, 3),
return_counts = True,
axis = 0)
添加回答
举报
0/150
提交
取消