【OpenCV经验总结】——视频篇

---

前言

本篇汇总了本人在OpenCV使用过程中总结的关于视频处理的一些函数使用经验及Bug处理经验，会根据后续使用情况逐步更新。

一、 视频读写

import cv2

# 读取视频文件
video = cv2.VideoCapture('./test.mp4')	# 参数为视频文件地址，若是数字表示摄像头编号。
'''
参数为字符串，表示输入的视频文件的地址及文件名
参数为数字，表示摄像头编号，默认为-1.即随机选取一个摄像头
'''

# 创建写视频器
video_writer = cv2.VideoWriter(filename='./output.mp4', 				# 保存路径文件名
							   apiPreference=cv2.CAP_FFMPEG,			# 后端
							   fourcc=cv2.VideoWriter_fourcc(*'mp4v'),	# 视频编解码器
							   fps=25,									# 视频帧率
							   frameSize=(1920,1080)					# 视频帧尺寸(W, H)
							   isColor=True								# 彩色图像或黑白图像
							   )
'''
apiPreference：cv2.CAP_FFMPEG 或者 cv2.CAP_GSTREAMER，此参数是3.x版本的opencv才有的
fourcc：cv2.VideoWriter_fourcc('M', 'P', '4', 'V') 或 fourcc=cv2.VideoWriter_fourcc(*'mp4v')
		支持类型：MP4V / X264 / I420 / PIMI / XVID / THEO / FLV1
'''							   

# 判断视频是否是打开状态
while video.isOpened():
	# 读取一帧
	ret, frame = video.read() # ret是bool类型，表示是否读取成功；frame为获取的帧图像

	if ret:
		# 写入一帧
		video_writer.write(frame)
		# 播放视频
		cv2.imshow('frame', frame)
		cv2.waitKey(1) # 通过设置等待时间改变播放速度
	else:
		break

# 释放
video.release()
video_writer.release()	# 不释放会无法完成写视频，类似文件写完后的close()

二、视频参数的获取和设置

import cv2

# 读取视频文件
video = cv2.VideoCapture('./test.mp4')

# 获取视频参数
param = video.get(CV_CAP_PROP_FRAME_COUNT) # 也可使用以下数字表示参数
'''
cv2.VideoCapture.get(0)		CV_CAP_PROP_POS_MSEC 		视频文件的当前位置，以毫秒为单位
cv2.VideoCapture.get(1)		CV_CAP_PROP_POS_FRAMES 		基于以0开始的被捕获或解码的帧索引
cv2.VideoCapture.get(2)		CV_CAP_PROP_POS_AVI_RATIO 	视频文件的相对位置（播放）：0=电影开始，1=影片的结尾。
cv2.VideoCapture.get(3)		CV_CAP_PROP_FRAME_WIDTH 	在视频流的帧的宽度
cv2.VideoCapture.get(4)		CV_CAP_PROP_FRAME_HEIGHT 	在视频流的帧的高度
cv2.VideoCapture.get(5)		CV_CAP_PROP_FPS 			帧速率
cv2.VideoCapture.get(6)		CV_CAP_PROP_FOURCC 			编解码器/fourcc
cv2.VideoCapture.get(7)		CV_CAP_PROP_FRAME_COUNT		帧数
cv2.VideoCapture.get(8)		CV_CAP_PROP_FORMAT 			返回对象的格式
cv2.VideoCapture.get(9)		CV_CAP_PROP_MODE 			返回后端特定的值，该值指示当前捕获模式
cv2.VideoCapture.get(10)	CV_CAP_PROP_BRIGHTNESS 		图像的亮度(仅适用于照相机)
cv2.VideoCapture.get(11)	CV_CAP_PROP_CONTRAST 		图像的对比度(仅适用于照相机)
cv2.VideoCapture.get(12)	CV_CAP_PROP_SATURATION 		图像的饱和度(仅适用于照相机)
cv2.VideoCapture.get(13)	CV_CAP_PROP_HUE 			色调图像(仅适用于照相机)
cv2.VideoCapture.get(14)	CV_CAP_PROP_GAIN 			图像增益(仅适用于照相机)（Gain在摄影中表示白平衡提升）
cv2.VideoCapture.get(15)	CV_CAP_PROP_EXPOSURE 		曝光(仅适用于照相机)
cv2.VideoCapture.get(16)	CV_CAP_PROP_CONVERT_RGB 	指示是否应将图像转换为RGB布尔标志
cv2.VideoCapture.get(17)	CV_CAP_PROP_WHITE_BALANCE 	白平衡，暂时不支持
cv2.VideoCapture.get(18)	CV_CAP_PROP_RECTIFICATION 	立体摄像机的矫正标注（目前只有DC1394 v.2.x后端支持这个功能）
'''

# 设置视频参数
video.set(CV_CAP_PROP_FPS, 30) # 第一个参数为设置的参数名，第二个参数为设定的值
'''
可设置的参数与上述可获取的参数大体一致
'''

# 释放
video.release()