如何用Python+OpenCV生成气泡检测用的MP4格式二值视频？

一、解决输出为Python源文件而非MP4的问题 #

你的原代码仅实现了实时显示二值化结果，没有添加视频保存逻辑，因此无法生成MP4格式文件。要输出MP4格式的二值视频，需通过OpenCV的VideoWriter类完成视频编码与写入，核心步骤如下：

1. 获取原视频的帧率、宽高参数
2. 初始化VideoWriter对象，明确指定输出路径（必须以.mp4结尾）、编码格式、帧率和分辨率
3. 在循环中将每帧二值化结果写入视频文件

二、优化二值视频清晰度的方法 #

结合气泡检测场景，可通过以下方式提升二值视频的清晰度与准确性：

• 调整背景减除算法参数，减少噪声干扰
• 添加形态学后处理，细化气泡边缘、消除杂点
• 保存时保持原视频分辨率与帧率，避免压缩损失

修改后的完整代码 #

import cv2

# 读取输入视频
cap = cv2.VideoCapture(r'C:\Users\Harshali Kulkarmi\OneDrive\Desktop\WASEDA UNI CLASSES\LAB WORK\Embolism research\videos\MAH00207.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))

# 初始化视频写入器：指定MP4编码、输出路径、帧率、分辨率
# 编码格式选择mp4v适配MP4容器，若出现编码问题可尝试'XVID'
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter('output_bubbles.mp4', fourcc, fps, (width, height), isColor=False)

# 初始化背景减除器：关闭阴影检测、调整阈值减少噪声
# history：背景模型的历史帧数，varThreshold：像素与背景的差异阈值
fgbg = cv2.createBackgroundSubtractorMOG2(history=500, varThreshold=40, detectShadows=False)

# 定义形态学操作核，用于消除噪声、细化气泡边缘
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # 应用背景减除得到二值前景掩码
    fgmask = fgbg.apply(frame)

    # 形态学处理：先腐蚀消除小噪声，再膨胀恢复气泡形状
    fgmask = cv2.morphologyEx(fgmask, cv2.MORPH_OPEN, kernel)

    # 将处理后的二值帧写入视频文件
    out.write(fgmask)

    # 实时显示结果
    cv2.imshow('Foreground Mask', fgmask)
    if cv2.waitKey(1) == ord('q'):
        break

# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()

关键说明 #

1. 视频保存逻辑：
   • 必须指定输出文件后缀为.mp4，否则会导致文件格式识别错误
   • isColor=False表示写入单通道二值图，若设置为True需将fgmask转换为三通道（cv2.cvtColor(fgmask, cv2.COLOR_GRAY2BGR)）
2. 清晰度优化细节：
   • detectShadows=False：关闭阴影检测，避免阴影被误判为前景气泡，减少二值图中的灰色杂点
   • 形态学开运算：先腐蚀后膨胀，有效消除背景减除产生的微小噪声点，同时保留气泡的完整形状
   • 保持原视频分辨率与帧率：确保输出视频的清晰度与原视频一致，避免缩放导致的模糊

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