相关文章推荐
强健的核桃  ·  dart - Flutter ios ...·  1 年前    · 
玩篮球的茴香  ·  java - Find the ...·  1 年前    · 
首发于 OpenCV

文章OpenCV02——读取视频,显示视频,保存视频

目标

在这里,将学习以下功能:

  • cv2.VideoCapture()
  • cv2.VideoWriter()

从相机中读取视频

通常情况下,用摄像机捕捉实时画面。提供了一个非常简单的界面。从摄像头捕捉一段视频(使用的是笔记本电脑内置的网络摄像头) ,将其转换成灰度视频并显示出来。只是一个简单的任务开始。要捕获视频,需要创建一个 VideoCapture 对象。它的参数可以是设备索引或视频文件的名称。设备索引就是指定哪个摄像头的数字。正常情况下,只有一个摄像头会被连接(就像我的情况一样)。所以简单地传0(或-1)。此外,如果有多个摄像头的话,可以通过传递1来选择第二个相机,以此类推。在此之后,你可以逐帧捕获。在最后记得进行释放。

import cv2
import numpy as np
cap = cv2.VideoCapture(0)  # 第一个摄像头
if not cap.isOpened():
 print("can not open camera")
 exit()
while True:
 # 逐帧捕获
 ret, frame = cap.read()
 # 如果正确读取帧,ret为True
 if not ret:
  print("Can't receive frame (stream end?). Exiting ...")
   break
  # 将当前帧转换为灰度图
  gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
  # 显示结果帧
  cv2.imshow('frame', gray)
  if cv2.waitKey(1) == ord('q'):
   break
 # 完成所有操作后,释放捕获器
cap.release()
cv2.destroyAllWindows()
  • cap.read() 返回布尔值( True / False )。如果正确读取了帧,它将为 True 。因此,你可以通过检查此返回值来检查视频的结尾。 有时,cap可能尚未初始化捕获。在这种情况下,此代码显示错误。你可以通过 cap.isOpened ()方法检查它是否已初始化。如果是 True ,那么确定是打开的。否则,使用 cap.open ()打开它。还可以使用 cap.get(propId) 方法访问该视频的某些功能,其中propId是0到18之间的一个数字。每个数字表示视频的属性(如果适用于该视频),并且可以显示完整的详细信息在这里看到: cv2::VideoCapture::get() 。其中一些值可以使用 cap.set(propId,value) 进行修改。 value是想要的新值。

从文件播放视频

它与从相机捕获相同,只是用视频文件名更改摄像机索引。另外,在显示框架时,请使用适当的时间 cv.waitKey() 。如果太小,则视频将非常快,但如果设置得太大,则视频将变得很慢。正常情况下25毫秒就可以了。

import cv2
import numpy as np
cap = cv2.VideoCapture('vtest.avi')
while cap.isOpened():
 ret, frame = cap.read()
 if not ret:
  print("Can't receive frame (stream end?). Exiting ...")
  break
 gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
 cv2.imshow('frame', gray)
 if cv2.waitKey(1) == ord('q'):
  break
cap.release()
cv2.destroyAllWindows()

保存视频

捕捉一个视频,一帧一帧地处理, 如果想要保存这个视频。对于图像,它非常简单,只需使用 cv2.imwrite() 。但是对于视频而言,这里还需要做一些工作。首先创建一个 VideoWriter 对象。应该指定输出文件名(例如: output.avi)。然后应该指定 FourCC 代码以及传递帧率的数量和帧大小。最后一个是颜色标志。如果为 True ,编码器期望颜色帧,否则它与灰度帧一起工作。FourCC: en.wikipedia.org/wiki/F 是用于指定视频编解码器的4字节代码。可用代码列表可在fourcc.org中: fourcc.org/codecs.php 找到。它取决于平台。遵循编解码器来说效果很好。

  • 在Fedora中:DIVX,XVID,MJPG,X264,WMV1,WMV2。(最好使用XVID。MJPG会生成大尺寸的视频。X264会生成非常小的尺寸的视频)
  • 在Windows中:DIVX(尚待测试和添加)
  • 在OSX中:MJPG(.mp4),DIVX(.avi),X264(.mkv)。

FourCC代码作为MJPG的 cv2.VideoWriter_fourcc('M','J','P','G') or cv2.VideoWriter_fourcc(*'MJPG') 传递。在从摄像机捕获的代码下面, 沿垂直方向翻转每一帧并保存

import cv2
import numpy as np
cap = cv2.VideoCapture(0)
#  定义编解码器并创建VideoWriter对象
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640, 480))
while cap.isOpend():
 ret, frame = cap.read()
 if not ret:
  print("Can't receive frame (stream end?). Exiting ...")
  break
 frame = cv2.flip(frame, 0)
 # 写翻转的框架
 out.write(frame)
  cv2.imshow('frame', frame)
  if cv2.waitKey(1) == ord('d'):