用matplotlib在二值化视频中央画一条线

以下是使用matplotlib在二值化视频中央画一条线的Python示例代码：

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

# 读取视频
cap = cv2.VideoCapture('video.mp4')

# 获取视频帧率和尺寸
fps = cap.get(cv2.CAP_PROP_FPS)
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

# 创建matplotlib图形窗口
fig, ax = plt.subplots()

# 循环读取视频帧
while cap.isOpened():
    ret, frame = cap.read()
    if ret:
        # 转换为灰度图像
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        
        # 二值化图像
        _, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)
        
        # 获取二值化图像中央行的像素值
        row = binary[height//2, :]
        
        # 获取中央行的白色像素位置
        white_pixels = np.where(row == 255)[0]
        
        # 计算中央行的中心位置
        if len(white_pixels) > 0:
            center = int(np.mean(white_pixels))
        else:
            center = width//2
        
        # 在图像上画一条竖直线
        ax.clear()
        ax.imshow(binary, cmap='gray')
        ax.axvline(center, color='r')
        ax.set_axis_off()
        plt.pause(1/fps)
    else:
        break

# 释放视频和关闭图形窗口
cap.release()
plt.close(fig)

该代码从视频文件中读取帧，并将每一帧转换为灰度图像和二值化图像。然后，它计算二值化图像中央行的白色像素位置，并计算中央行的中心位置。最后，它在matplotlib图形窗口上画一条竖直线，表示中央行的中心位置。代码使用`plt.pause()`函数来控制图形窗口的显示速度，以匹配视频帧率。