opencv-python 摄像头手动截图小脚本、视频自动截图

摄像头手动截图

# coding:utf-8
import cv2
import os
import time

# 网络摄像头
# url = http://admin:admin@192.168.31.117:8081

# 海康摄像头
# url = 'rtsp://admin:tsg610@192.168.31.57:554//Streaming/Channels/2'

# 本地摄像头
url = 0

cap = cv2.VideoCapture(url)
index = 1
# 是否存在jpg目录
if os.path.exists("jpg") == False:
    # 新建一个保存图像的目录
    os.mkdir("jpg")

while cap.isOpened():
    ret, frame = cap.read()
    cv2.imshow("Capture_Paizhao", frame)

    k = cv2.waitKey(1) & 0xFF
    if k == ord('s'):  # 按下s键，进入下面的保存图片操作
        # 图像保存在jpg文件夹内  名字：程序运行第几张 + 当前时间戳 + .jpg
        cv2.imwrite("jpg/" + str(index) + "_" + str(int(time.time())) + ".jpg", frame)

        print("图像宽高", int(cap.get(3)), int(cap.get(4)))
        print("已保存" + str(index) + ".jpg")
        print("-" * 30)

        index += 1
    elif k == ord('q'):  # 按下q键，程序退出
        break
cap.release()
cv2.destroyAllWindows()

视频自动截图

# coding:utf-8
import cv2
import os

# 网络摄像头
# url = http://admin:admin@192.169.31.119:9091

# 海康摄像头
# url = 'rtsp://admin:tsg610@192.169.31.59:554//Streaming/Channels/2'

# 视频文件名
o = "18"
# 摄像头1秒多少帧数 * 几秒
n, m = 30, 10
# 视频后缀
cap = cv2.VideoCapture(o + ".mp4")

# 是否存在jpg目录
if os.path.exists("jpg_" + o) == False:
    # 新建一个保存图像的目录
    os.mkdir("jpg_" + o)

l = 0
index = 0
while cap.isOpened():
    ret, frame = cap.read()
    l += 1

    if not ret:
        break

    # 多少帧数 * 秒
    if l == (n * m):
        l = 0
        # 图像保存在jpg文件夹内  名字：程序运行第几张 + 当前时间戳 + .jpg
        cv2.imwrite("jpg_" + o + "/" + "q0" + o + "_" + str(index) + ".jpg", frame)

        print("图像宽高", int(cap.get(3)), int(cap.get(4)))
        print("已保存" + str(index) + ".jpg")
        print("-" * 30)

        index += 1

cap.release()
cv2.destroyAllWindows()