mean_color = cv2.mean(img, mask=mask)[:3] cv2.error: OpenCV(4.5.5) /Users/runner/work/opencv-python/opencv-python/opencv/modules/core/src/mean.dispatch.cpp:126: error: (-215:Assertion failed) mask.empty() || mask.type() == CV_8U in function 'mean'报错了
时间: 2024-01-16 07:02:02 浏览: 219
抱歉,这是因为在调用cv2.mean()函数时,mask参数需要传入一个与img形状相同的掩膜图像,用于计算掩膜区域内的像素值平均值。而在上述代码中,mask变量是通过np.zeros_like(img)创建的全零掩膜图像,需要将其处理成与img形状相同的掩膜图像才能正确计算平均值。
以下是修改后的代码,可以正常计算出最大颜色:
```python
import cv2
import numpy as np
# 读取图片
img = cv2.imread('pig.jpg')
# 将图片转为灰度图
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 二值化分割
ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
# Laplacian算法提取轮廓
laplacian = cv2.Laplacian(binary, cv2.CV_8U)
contours, hierarchy = cv2.findContours(laplacian, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 检测最大颜色
max_color = None
max_count = 0
for contour in contours:
mask = np.zeros_like(gray)
cv2.drawContours(mask, [contour], 0, 255, -1)
mean_color = cv2.mean(img, mask=mask)[:3]
count = cv2.countNonZero(mask)
if count > max_count:
max_color = mean_color
max_count = count
# 输出最大颜色
print('The max color is:', max_color)
# 显示图片
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
需要注意的是,这里的掩膜图像mask是使用np.zeros_like(gray)创建的,对应的是灰度图像gray,而绘制轮廓时需要使用cv2.drawContours()函数来绘制轮廓所在的区域。同时,在计算轮廓区域的像素个数时,需要使用cv2.countNonZero()函数。
阅读全文