OpenCV读取NumPy数组报错：霍夫线检测通道数异常问题

错误根源 #

1. 通道转换逻辑错误：你调用cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)时，输入的img是单通道数组（shape为(50,50)），但COLOR_BGR2GRAY仅适用于3通道BGR图像转灰度，OpenCV检测到输入通道数为1，不符合该转换的要求，因此抛出通道数异常。
2. 数据类型不兼容：你生成的数组是int64类型，而OpenCV处理图像默认使用uint8（8位无符号整数）格式，类型不匹配会导致后续操作异常。
3. 参数与图像尺寸不匹配：原代码中minLineLength=100远大于图像尺寸（50x50），无法检测到任何线条；同时单通道图像无法绘制彩色BGR线条。

修正后的代码 #

import numpy as np
import cv2
from matplotlib import pyplot as plt

img_size = 50
# 生成符合OpenCV要求的uint8类型单通道灰度图像（像素值0/255）
img = np.random.choice([0, 255], size=(img_size, img_size), p=[1/30, 29/30]).astype(np.uint8)
plt.imshow(img, cmap='gray', interpolation='nearest')
plt.show()

# 直接使用单通道灰度图做边缘检测，无需额外转灰度
edges = cv2.Canny(img, 50, 150, apertureSize=3)
# 调整霍夫线检测参数，适配小尺寸图像
minLineLength = 10
maxLineGap = 2
lines = cv2.HoughLinesP(edges, 1, np.pi/180, threshold=5, minLineLength=minLineLength, maxLineGap=maxLineGap)

# 将单通道图像转为3通道BGR格式，才能绘制彩色线条
img_color = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)

# 仅当检测到线条时才执行绘制，避免索引错误
if lines is not None:
    for line in lines:
        x1, y1, x2, y2 = line[0]
        cv2.line(img_color, (x1, y1), (x2, y2), (0, 255, 0), 1)

cv2.imshow('Detected Lines', img_color)
cv2.waitKey(0)
cv2.destroyAllWindows()

关键修正点 #

• 移除无效的灰度转换：输入数组本身就是单通道灰度图，无需调用COLOR_BGR2GRAY转换。若需要彩色绘图，使用COLOR_GRAY2BGR将单通道转为3通道。
• 统一数据类型：将数组转为np.uint8，并将像素值调整为0/255（符合8位灰度图像的标准取值范围）。
• 适配图像尺寸调整参数：针对50x50的小图像，缩小minLineLength和maxLineGap的取值，确保能检测到线条；添加lines is not None判断，避免空指针错误。
• 彩色绘图兼容处理：单通道图像无法显示彩色，必须先转为3通道BGR格式后再绘制彩色线条。

内容的提问来源于stack exchange，提问作者fynn