之前一直不清楚opencv调用摄像头的具体方法,于是参考b站视频写了个通用模板,以后可以根据需要自行调整。视频链接
先定义图片和视频的保存路径
img_path = './img_video/test_img.jpg' # 图片保存路径
video_path = './img_video/test_video.mp4' # 视频保存路径
这个是图片处理的函数,由摄像头捕获的图片或视频帧可以通过这个函数处理后再返回,我这里没有作处理直接返回。
def process_frame(frame):
return frame
捕获单张图片并保存
# 延迟2秒
time.sleep(2)
# 调用摄像头,0是默认摄像头,1是外置摄像头
cap = cv2.VideoCapture(0)
# 捕获并处理一帧画面
success, frame = cap.read()
if not success:
print('error!')
frame = process_frame(frame)
# 关闭摄像头
cap.release()
# 关闭窗口
cv2.destroyAllWindows()
# 保存图片
cv2.imwrite(img_path, frame)
print('图片已保存', img_path)
实时显示视频并保存(捕获视频其实也就是循环捕获图片)
# 调用摄像头,0是默认摄像头,1是外置摄像头
cap = cv2.VideoCapture(0)
# 打开cap
cap.open(0)
#视频尺寸
frame_size = (cap.get(cv2.CAP_PROP_FRAME_WIDTH),
cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
#文件编码方式
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = cap.get(cv2.CAP_PROP_FPS)
out = cv2.VideoWriter(video_path, fourcc, fps,
(int(frame_size[0]), int(frame_size[1])))
# --!!!关键部分,无限循环--
while cap.isOpened():
# 获取画面
success, frame = cap.read()
# 未捕获到,报错并退出
if not success:
print('error!')
break
# 对每帧进行处理
frame = process_frame(frame)
# 将处理后的帧写入视频文件
out.write(frame)
# 实时显示图像
cv2.imshow('press q to quit', frame)
# 按q或esc退出
if cv2.waitKey(1) in [ord('q'), 27]:
break
# 关闭图像窗口
cv2.destroyAllWindows()
out.release()
# 关闭摄像头
cap.release()
print('视频已保存', video_path)
完整代码:(需要在main里自行调整捕获图片还是视频)
import cv2
import matplotlib.pyplot as plt
import time
img_path = './img_video/test_img.jpg' # 图片保存路径
video_path = './img_video/test_video.mp4' # 视频保存路径
# 自定义处理图像函数,也可以不处理直接返回
def process_frame(frame):
return frame
# 调用摄像头拍摄照片并保存
def get_img():
# 延迟2秒
time.sleep(2)
# 调用摄像头,0是默认摄像头,1是外置摄像头
cap = cv2.VideoCapture(0)
# 捕获并处理一帧画面
success, frame = cap.read()
if not success:
print('error!')
frame = process_frame(frame)
# 关闭摄像头
cap.release()
# 关闭窗口
cv2.destroyAllWindows()
# 保存图片
cv2.imwrite(img_path, frame)
print('图片已保存', img_path)
# 调用摄像头捕获并保存视频
def get_video():
# 调用摄像头,0是默认摄像头,1是外置摄像头
cap = cv2.VideoCapture(0)
# 打开cap
cap.open(0)
#视频尺寸
frame_size = (cap.get(cv2.CAP_PROP_FRAME_WIDTH),
cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
#文件编码方式
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = cap.get(cv2.CAP_PROP_FPS)
out = cv2.VideoWriter(video_path, fourcc, fps,
(int(frame_size[0]), int(frame_size[1])))
# --!!!关键部分,无限循环--
while cap.isOpened():
# 获取画面
success, frame = cap.read()
# 未捕获到,报错并退出
if not success:
print('error!')
break
# 对每帧进行处理
frame = process_frame(frame)
# 将处理后的帧写入视频文件
out.write(frame)
# 实时显示图像
cv2.imshow('press q to quit', frame)
# 按q或esc退出
if cv2.waitKey(1) in [ord('q'), 27]:
break
# 关闭图像窗口
cv2.destroyAllWindows()
out.release()
# 关闭摄像头
cap.release()
print('视频已保存', video_path)
def main():
#get_img()
get_video()
if __name__ == '__main__':
main()