使用ImageAI实时目标检测教程时遭遇OpenCV读取错误求助

错误原因分析 #

你遇到的!_filename.empty()错误，看似是文件路径问题，但实际是以下几个可能的原因：

• 摄像头初始化失败，导致cv2.VideoCapture(0)返回的对象无效，ImageAI内部误将其当作空路径处理
• 输出视频路径未指定扩展名，OpenCV无法识别输出格式，引发内部逻辑错误
• ImageAI版本兼容性问题，旧版本对摄像头输入的处理存在bug

分步解决方案 #

1. 先验证摄像头是否正常工作 #

首先排除硬件或OpenCV本身的问题，运行以下测试代码：

import cv2

# 初始化摄像头
camera = cv2.VideoCapture(0)

# 检查摄像头是否成功打开
if not camera.isOpened():
    print("无法打开摄像头，请检查硬件连接或权限")
else:
    print("摄像头打开成功，按q退出预览")
    while True:
        ret, frame = camera.read()
        if ret:
            cv2.imshow("摄像头预览", frame)
            # 按q键退出
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
    # 释放资源
    camera.release()
    cv2.destroyAllWindows()

如果这段代码无法显示摄像头画面，说明是摄像头权限或OpenCV配置问题，需要先解决这个基础问题。

2. 修正输出视频路径的扩展名 #

你的代码中output_file_path没有指定视频扩展名，OpenCV无法确定输出格式，这会触发内部错误。修改路径为带扩展名的格式，比如：

video_path = detector.detectObjectsFromVideo(
    camera_input=camera,
    output_file_path=os.path.join(execution_path, "camera_detected_video.avi"),
    frames_per_second=20,
    log_progress=True,
    minimum_percentage_probability=30
)

常用的视频扩展名有.avi、.mp4，根据你的OpenCV支持情况选择。

3. 升级ImageAI到兼容版本 #

如果上述方法无效，可能是ImageAI版本的bug，执行以下命令升级：

pip install --upgrade imageai

同时确保你的TensorFlow版本与ImageAI兼容（ImageAI v3.x支持TensorFlow 2.x，v2.x支持TensorFlow 1.x）。

4. 实时预览替代保存视频（可选） #

如果你的需求是实时检测并显示画面，而非保存视频，可以使用自定义回调函数，避免输出路径的问题：

from imageai.Detection import VideoObjectDetection
import os
import cv2

def forFrame(frame_number, output_array, output_count, frame):
    # 在画面上绘制检测结果
    for obj in output_array:
        cv2.rectangle(frame, (obj["box_points"][0], obj["box_points"][1]), 
                      (obj["box_points"][2], obj["box_points"][3]), (0, 255, 0), 2)
        cv2.putText(frame, f"{obj['name']}: {obj['percentage_probability']:.1f}%", 
                    (obj["box_points"][0], obj["box_points"][1]-10), 
                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
    # 显示画面
    cv2.imshow("实时目标检测", frame)
    # 按q退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        raise Exception("用户退出")

execution_path = os.getcwd()
camera = cv2.VideoCapture(0)

detector = VideoObjectDetection()
detector.setModelTypeAsYOLOv3()
detector.setModelPath(os.path.join(execution_path , "yolov3.pt"))
detector.loadModel()

try:
    detector.detectObjectsFromVideo(
        camera_input=camera,
        per_frame_function=forFrame,
        frames_per_second=20,
        minimum_percentage_probability=30
    )
except Exception as e:
    camera.release()
    cv2.destroyAllWindows()

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