拉不拉稀肚拉稀 发表于 2024-8-6 20:36:11

Python:一个单针刻度表的辨认步伐

辨认单针刻度表(例如钟表)的时间通常涉及几个关键步调,包罗图像预处理、针检测、刻度辨认和时间盘算。以下是一个示例方案,展示如何使用 OpenCV 和 Python 进行单针刻度表的辨认。假设我们处理的是简化的钟表图像,其中针清晰可见且配景简朴。
1. 情况设置

确保已安装必要的库:
pip install opencv-python numpy matplotlib
2. 代码示例

以下是一个根本示例代码,演示如何检测单针刻度表的针,并盘算出指示的时间:
import cv2
import numpy as np
import math
import matplotlib.pyplot as plt

def preprocess_image(image_path):
    # 读取图像
    img = cv2.imread(image_path)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    # 图像去噪
    blurred = cv2.GaussianBlur(gray, (5, 5), 0)

    # 边缘检测
    edges = cv2.Canny(blurred, 50, 150)

    return img, gray, edges

def detect_dial_contour(edges):
    # 检测圆形轮廓
    circles = cv2.HoughCircles(edges, cv2.HOUGH_GRADIENT, 1, 20,
                               param1=50, param2=30, minRadius=10, maxRadius=100)

    if circles is not None:
      circles = np.round(circles).astype("int")
      return circles
    return None

def detect_hand(gray, circles):
    if circles is None:
      return None, None

    # 选择最大的圆作为表盘
    (cX, cY, cR) = circles
    mask = np.zeros_like(gray)
    cv2.circle(mask, (cX, cY), cR, 255, -1)
    masked_gray = cv2.bitwise_and(gray, gray, mask=mask)

    # 找到针
    _, thresh = cv2.threshold(masked_gray, 1, 255, cv2.THRESH_BINARY)
    contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    if contours:
      # 假设最大的轮廓是指针
      contour = max(contours, key=cv2.contourArea)
      M = cv2.moments(contour)
      if M["m00"] != 0:
            cX = int(M["m10"] / M["m00"])
            cY = int(M["m01"] / M["m00"])
            return (cX, cY), (cX, cY)
    return None, None

def calculate_angle(hand_point, center):
    dx = hand_point - center
    dy = hand_point - center
    angle = math.atan2(dy, dx) * 180 / np.pi
    if angle < 0:
      angle += 360
    return angle

def main(image_path):
    img, gray, edges = preprocess_image(image_path)
    circles = detect_dial_contour(edges)
    hand_point, center = detect_hand(gray, circles)
   
    if hand_point is not None:
      # Calculate angle
      angle = calculate_angle(hand_point, center)
      print(f"Detected angle: {angle:.2f} degrees")
      
      # Draw results
      cv2.circle(img, center, 5, (0, 255, 0), -1)
      cv2.line(img, center, hand_point, (0, 0, 255), 2)
      cv2.imshow("Clock", img)
      cv2.waitKey(0)
      cv2.destroyAllWindows()
    else:
      print("No hand detected")

if __name__ == "__main__":
    main("clock.jpg")
3. 代码表明


[*] 图像预处理:

[*]将图像转换为灰度图像。
[*]使用高斯模糊和边缘检测来预备后续的处理。

[*] 检测刻度表轮廓:

[*]使用霍夫圆变换检测钟表的外圈。
[*]在检测到的圆上创建一个掩码,将其他地区清除在外。

[*] 检测指针:

[*]在掩码地区内检测轮廓,假设最大的轮廓为指针。
[*]盘算指针的中心点。

[*] 盘算角度:

[*]根据指针的方向盘算角度。

[*] 显示效果:

[*]绘制圆心、指针和检测效果,显示图像。

4. 扩展和改进



[*]复杂配景处理:对于复杂配景,可以使用更多的预处理步调(如颜色空间转换或形态学操作)来增强图像质量。
[*]指针长度和角度盘算:如果指针比较短,可能需要改进指针检测算法。
[*]多针表盘:如果表盘上有多根指针,可能需要使用更复杂的算法来区分时针、分针和秒针。
总结

这个示例展示了如何使用 OpenCV 和 Python 来辨认单针刻度表的时间。你可以根据具体的钟表图像和需求进行调整和优化。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: Python:一个单针刻度表的辨认步伐