多进程截取海康相机视频帧存储

前言

使用IP拼写RTSP进行连接海康摄像头，我们需要知道该相机的IP号、账号、密码、端口号。但是过多的摄像头，我们就需要依靠MySql存储相应的摄像机的信息，然后再从MySql中调取使用Opencv读取视频流......

在这里我分享目前我做的一个版本的多进程式（采用进程池）连接摄像头截取视频帧。 基本流程：

1. 从数据库中获取ip, username, password

2. 组合ip, username, password的LIST

3. SaveFrams模块，从list中提取相应的ip, username, password

4. 进程池构造结构镶嵌第三步

一.连接数据库获取相机参数模块

导入pymysql模块，构建一个Class,便于调用。

1.1 使用conn = pymysql.connect确定好输入连接数据库基础参数（host、user、passwd、port、db和charset）

1.2 生成游标对象self.cur = self.conn.cursor()

1.3 Sql语句查询 ip,username, password

二. 获取 ip, username, password模块

定义函数getInfolIST。

2.1 调用第一步中构造好的类：Config = BaseSet()

2.2 分别设置空的list( IPlIST = [],usernamelIST = [],passwordlIST = [])

2.3 构造for in循环，将数据库中的数据组合成相应的list

2.4 return 三个list便于调用

三. 存储图像模块

从上面的ip, username, password按照如下规则拼写RTSP流：

"rtsp://%s:%s@%s:554/cam/realmonitor?channel=1&subtype=1" % (username, password, ip)

3.1 构造SaveFrams(username, password, ip)函数

3.2 设置图像存储路径

3.3 拼写RTSP（从输入中获取）

3.4 使用OpenCv读取RTSP

3.5 判断RTSP是否是可打开，若打开则接下执行

3.6  cap.read()获取frame

3.7 使用cv2.resize调整自己需要图的大小

3.8 使用cv2.imwrite 存储自己的图像（命名规则：ip+uuid;图像格式为JPG）

3.9 释放cap: cap.release()

四.多进程模块

先调用 获取 ip, username, password模块获取相应的IPlIST, usernamelIST, passwordlIST，设置进程池（这里大家可以控制进程池数），经过一个for in 循环 分别从list取出相应的ip, username, password，传送给存储图像模块。

if __name__ == '__main__':
    IPlIST, usernamelIST, passwordlIST = getInfolIST()
    try:
        while True:
            time.sleep(1)
            p = multiprocessing.Pool(32)
            r_list = []
            for username, password, ip in zip(usernamelIST, passwordlIST, IPlIST):
                r_list.append(p.apply_async(SaveFrams, args=(username, password, ip)))
            p.close()
            for r in r_list:
                r.get()
    except KeyboardInterrupt:
        p.terminate()
        print('[-] main exit')
    p.join()