python - 使用 numpy(快速)计算平均栅格值

Question

我有一个 JPEG 图像，我需要尽快计算每个栅格(红色、蓝色和绿色)的平均值。当我尝试使用两个 for 循环访问每个像素并添加值时，该过程需要很长时间(大约 30 秒)。有没有办法快速计算平均栅格值(可能使用 numpy 和 OpenCV)？我使用的代码如下:

for i in range(width):
        for j in range(height):
            pix = im.getpixel((i,j))
            redValues = redValues + pix[0]
            greenValues = greenValues + pix[1]
            blueValues = blueValues + pix[2]

Answer 1

import cv2 

## Read the image into `BGR`: (H,W,3), [[B,G,R],...]
img = cv2.imread("test.png")

## Get shape
H,W = img.shape[:2]

## calculate 
avg = img.reshape(-1,3).sum(axis=0)/(H*W)
print(avg)