如何用Python的OpenCV检测药盒单元格轮廓？低质相机优化建议

嘿，我之前也碰到过树莓派相机画质拉胯导致边缘检测翻车的情况，给你几个亲测有效的思路，应该能帮你搞定药盒单元格的轮廓检测：

1. 优先做针对性的图像预处理 #

低画质图像的核心问题是噪点多、亮度不均，先把图像“清理干净”再做检测会轻松很多：

• 高斯模糊去噪：用cv2.GaussianBlur()平滑图像，过滤掉树莓派相机常见的高频噪点，同时不会破坏单元格的大轮廓：

  gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
  # 核大小根据图像分辨率调整，(5,5)适合多数树莓派拍摄的画面
  blurred = cv2.GaussianBlur(gray, (5,5), 0)
  

• 自适应阈值分割：代替全局阈值，解决低光照下的亮度不均问题，让单元格边界更清晰：

  # 自适应高斯阈值，反二值化让单元格轮廓为白色
  thresh = cv2.adaptiveThreshold(blurred, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 11, 2)
  

• 形态学操作补全轮廓：用闭运算填补轮廓的小缺口，让单元格的边界更连贯：

  # 矩形核适合规整的单元格轮廓
  kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (3,3))
  processed = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel)
  

2. 用轮廓检测替代霍夫变换 #

霍夫变换更适合检测孤立直线，而单元格是规整的矩形区域，直接提取轮廓并过滤更高效：

• 提取并过滤轮廓：用cv2.findContours()获取所有轮廓，然后通过面积、形状筛选出单元格：

  contours, _ = cv2.findContours(processed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
  cell_contours = []
  for cnt in contours:
      area = cv2.contourArea(cnt)
      # 根据你的药盒实际大小调整面积阈值，过滤掉噪点和过大的轮廓
      if 500 < area < 5000:
          # 用多边形近似判断是否为四边形（单元格的矩形特征）
          peri = cv2.arcLength(cnt, True)
          approx = cv2.approxPolyDP(cnt, 0.02 * peri, True)
          if len(approx) == 4:
              cell_contours.append(approx)
  

• 先定位药盒整体轮廓：如果药盒在画面中是完整的，先找到药盒的大轮廓，再在内部划分网格，能避免外界干扰，更准确地定位14个单元格。

3. 尝试颜色空间分割 #

如果灰度图的对比度不够，试试切换到HSV或LAB颜色空间，利用药盒和药品的颜色差异先提取目标区域：

hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# 用取色工具获取药盒颜色的HSV上下限，这里示例为浅色调药盒
lower_color = np.array([0, 0, 200])
upper_color = np.array([180, 25, 255])
# 生成颜色掩码，只保留药盒区域
mask = cv2.inRange(hsv, lower_color, upper_color)
# 对掩码做形态学操作后再进行轮廓检测

4. 硬件层面优化（减轻软件压力） #

有时候调整相机参数和补光，比纯软件处理效果更好：

• 调整树莓派相机参数：提高对比度、饱和度，关闭自动曝光（光线稳定场景下）：

  from picamera import PiCamera
  camera = PiCamera()
  camera.contrast = 60  # 提高对比度增强边界
  camera.saturation = 40  # 强化颜色差异
  camera.exposure_mode = 'off'  # 固定曝光避免亮度波动
  

• 增加均匀补光：低画质很大程度源于光线不足，加个小型LED补光灯，让药盒表面光照均匀，减少阴影，能大幅提升边缘检测的准确性。

你可以先从预处理步骤开始调试，逐步优化参数，应该能解决低画质带来的边缘检测问题。

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