今天用想用分类后的图像,统计其中不同类别像素数,也就是图像上有红红绿绿的像素,统计其中红色的像素个数,绿色的像素个数等等,直观的想法是遍历图像的每一个像素,然后一个一个计数,然而我要用python去做,这样python去循环遍历是很慢的,而且如果图像很大,那么耗时是不能被接受的,所以想到python的矩阵操作应该可以解决这个问题。
1、参考了numpy的布尔索引: 用布尔索引找到像素值为红色,或绿色的像素,然后在求返回来的矩阵的大小就知道有多少个像素了。
g = mask_full_binarization[:,:,1] == 255
r = mask_full_binarization[:,:,0] == 255
greenpixel_count = len(mask_full_binarization[g])
redpixel_count = len(mask_full_binarization[r])
2、另一种方法是经过一位大神指点的,用到numpy的广播机制,代码如下:
greenpixel_count = (mask_full_binarization.reshape(-1, 3)[:,1] == 255).sum()
redpixel_count = (mask_full_binarization.reshape(-1, 3)[:,0] == 255).sum()
大神说好像也可以用numpy的histogram,目前还不了解,以后有时间用到会进一步了解。