Python+Opencv根据URL读取视频流(摄像头)

Python+Opencv根据URL读取视频流(摄像头)

import os
import cv2
import pandas as pd

path = '/home/lxz/pytorch-YOLOv4-master/cam.xlsx'
save_path = 'result'

data = pd.DataFrame(pd.read_excel(path))#使用DataFrame读取表格
ids = data['id']#获取表格中id列的所有数据

for id in ids:
    res = data.loc[data['id'] == id, 'url']#id列中每个id名字对应的url地址
    res = list(res)#进一步转化为列表
    url = res[0]#取url那一列

    cap = cv2.VideoCapture(url)#设置视频流容器

    count = 0
    while count < 721:
        success, frame = cap.read()#读取视频流(successs代表读取成功,为True;frame为读取图片)
        if success:
            cv2.imshow('rea', frame)
            cv2.waitKey(5)
            if count % 24 == 1:#每24帧取一次图片
                img_name = str(id) + '_' + str(count) + '.jpg'#对输出的图片加入每个id名字命名
                cv2.imwrite(os.path.join(save_path, img_name), frame)
        count += 1
    cap.release()

pandas表格操作:想知道就点我哦

你可能感兴趣的:(Python+Opencv根据URL读取视频流(摄像头))