如何使用python和Opencv阅读视频文件

Question

我正在阅读一个avi文件usinh python 2.7和opencv2.4.我使用的是Windows 10.我的示例代码是

import numpy as np
import cv2
cap = cv2.VideoCapture('videos/wa.avi')
while(cap.isOpened()):
  ret, frame = cap.read()
  gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
  cv2.imshow('frame',gray)
  if cv2.waitKey(1) & 0xFF == ord('q'):
    break

cap.release()
cv2.destroyAllWindows()

当我运行视频显示但程序结束时没有错误

OpenCV Error: Assertion failed (size.width>0 && size.height>0) in cv::imshow, file ..\..\..\..\opencv\modules\highgui\src\window.cpp, line 261
Traceback (most recent call last):
File "C:/Users/Emmanu/PycharmProjects/VideoEventDetection/test.py", line 11, in <module>
cv2.imshow('frame',frame)
cv2.error: ..\..\..\..\opencv\modules\highgui\src\window.cpp:261: error: (-215) size.width>0 && size.height>0 in function cv::imshow

我做错了什么？我怎么能纠正它？

Answer 1

问题出在这一行:

gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

这一行期望frame是一个3通道或4通道Mat对象,但它得到了一些空Mat,这就是为什么你得到这个断言失败的原因.您需要检查视频中是否存在帧,并且需要正确处理视频结束.

cap.isOpened() 只检查是否可以打开视频文件进行读取,但是当达到视频文件结束时它不会返回错误.

试试这个