python多线程，多进程 队列处理图像识别

#多进程
from multiprocessing import Queue, Process, freeze_support
import cv2
import dlib,numpy

# 1.人脸关键点检测器
predictor_path = "shape_predictor_68_face_landmarks.dat"
# 2.人脸识别模型
face_rec_model_path = "dlib_face_recognition_resnet_model_v1.dat"
# 3.候选人文件
candidate_npydata_path = "candidates.npy"
candidate_path = "candidates.txt"
# 4.储存截图目录
path_screenshots = "screenShots/"


# 加载正脸检测器
detector = dlib.get_frontal_face_detector()
# 加载人脸关键点检测器
sp = dlib.shape_predictor(predictor_path)
# 加载人脸识别模型
facerec = dlib.face_recognition_model_v1(face_rec_model_path)


# 候选人脸描述子list
# 读取候选人数据
npy_data=numpy.load(candidate_npydata_path)
descriptors=npy_data.tolist()
# 候选人名单
candidate = []
file=open(candidate_path, 'r')
list_read = file.readlines()
for name in list_read:
    name = name.strip('\n')
    candidate.append(name)


def inputQ(queue):
    cv2.namedWindow("camera", 1)
    cap = cv2.VideoCapture(0)
    while True:
        ret, frame = cap.read()
        if ret == True:
            cv2.imshow('camera', frame)
            queue.put(frame)
            print(queue.qsize())
        else:
            break
        if (cv2.waitKey(1)) == ord('q'):
            break

    cap.release()
    cv2.destroyAllWindows()


def outputQ(queue,queue_out):
    while True:
        img = queue.get()
        cv2.putText(img, "press 'S': screenshot", (20, 400), cv2.FONT_HERSHEY_PLAIN, 1, (255, 255, 255), 1, cv2.LINE_AA)
        cv2.putText(img, "press 'Q': quit", (20, 450), cv2.FONT_HERSHEY_PLAIN, 1, (255, 255, 255), 1, cv2.LINE_AA)

        # img_gray = cv2.cvtColor(im_rd, cv2.COLOR_RGB2GRAY)
        dets = detector(img, 1)
        if len(dets) != 0:
            # 检测到人脸
            for k, d in enumerate(dets):
                # 关键点检测
                shape = sp(img, d)
                # 遍历所有点圈出来
                for pt in shape.parts():
                    pt_pos = (pt.x, pt.y)
                    cv2.circle(img, pt_pos, 2, (0, 255, 0), 1)
                face_descriptor = facerec.compute_face_descriptor(img, shape)
                d_test2 = numpy.array(face_descriptor)
                # 计算欧式距离
                dist = []
                for i in descriptors:
                    dist_ = numpy.linalg.norm(i - d_test2)
                    dist.append(dist_)
                num = dist.index(min(dist))  # 返回最小值

                left_top = (dlib.rectangle.left(d), dlib.rectangle.top(d))
                right_bottom = (dlib.rectangle.right(d), dlib.rectangle.bottom(d))
                cv2.rectangle(img, left_top, right_bottom, (0, 255, 0), 2, cv2.LINE_AA)
                text_point = (dlib.rectangle.left(d), dlib.rectangle.top(d) - 5)
                cv2.putText(img, candidate[num][0:4], text_point, cv2.FONT_HERSHEY_PLAIN, 2.0, (255, 255, 255), 1, 1)  # 标出face

            cv2.putText(img, "facesNum: " + str(len(dets)), (20, 50),  cv2.FONT_HERSHEY_PLAIN, 1.5, (0, 0, 0), 2, cv2.LINE_AA)
        else:
            # 没有检测到人脸
            cv2.putText(img, "facesNum:0", (20, 50),  cv2.FONT_HERSHEY_PLAIN, 1.5, (0, 0, 0), 2, cv2.LINE_AA)
        queue_out.put(img)


def show(queue_out):
    cv2.namedWindow("out", 1)
    while True:
        frame = queue_out.get()
        cv2.imshow('out', frame)
        if (cv2.waitKey(1)) == ord('q'):
            break
    cv2.destroyAllWindows()



if __name__ == '__main__':
    freeze_support()
    record1 = [] #store input process
    record2 = [] #stroe output process
    queue = Queue()
    queue_out = Queue()

    # 输入进程
    for i in range(1):
        process = Process(target=inputQ, args=(queue,))
        process.start()
        record2.append(process)
    #输出进程
    for i in range(3):
        process = Process(target=outputQ, args=(queue,queue_out))
        process.start()
        record2.append(process)
    #显示进程
    for i in range(1):
        process = Process(target=show, args=(queue_out,))
        process.start()
        record2.append(process)

    for p in record1:
        p.join()
    for p in record2:
        p.join()

#多线程
# -*- coding: UTF-8 -*-
import threading
import time
import queue
import dlib,numpy
import cv2
from concurrent.futures import ThreadPoolExecutor



url="rtsp://admin:asdf1234@192.168.31.103/Streaming/Channels/1"

# 1.人脸关键点检测器
predictor_path = "shape_predictor_68_face_landmarks.dat"
# 2.人脸识别模型
face_rec_model_path = "dlib_face_recognition_resnet_model_v1.dat"
# 3.候选人文件
candidate_npydata_path = "candidates.npy"
candidate_path = "candidates.txt"
# 4.储存截图目录
path_screenshots = "screenShots/"

q = queue.Queue()
q_show = queue.Queue()
# 加载正脸检测器
detector = dlib.get_frontal_face_detector()
# 加载人脸关键点检测器
sp = dlib.shape_predictor(predictor_path)
# 加载人脸识别模型
facerec = dlib.face_recognition_model_v1(face_rec_model_path)


# 候选人脸描述子list
# 读取候选人数据
npy_data=numpy.load(candidate_npydata_path)
descriptors=npy_data.tolist()
# 候选人名单
candidate = []
file=open(candidate_path, 'r')
list_read = file.readlines()
for name in list_read:
    name = name.strip('\n')
    candidate.append(name)


# 截图 screenshots 的计数器
cnt = 0
num = 0

def input_q():
    cv2.namedWindow("camera", 1)
    # cap = cv2.VideoCapture(url)
    cap = cv2.VideoCapture(0)

    while True:
        ret, frame = cap.read()
        if ret == True:
            cv2.imshow('camera', frame)
            q.put(frame)
            print(q.qsize())
        else:
            break

        # handle for the video

        # handle for exit
        if (cv2.waitKey(1)) == ord('q'):
            break

    cap.release()
    cv2.destroyAllWindows()

def output():
    # cv2.namedWindow(name, 1)
    # img_gray = cv2.cvtColor(im_rd, cv2.COLOR_RGB2GRAY)
    while True:
        print(q.qsize())
        img = q.get()
        cv2.putText(img, "press 'S': screenshot", (20, 400), cv2.FONT_HERSHEY_PLAIN, 1, (255, 255, 255), 1, cv2.LINE_AA)
        cv2.putText(img, "press 'Q': quit", (20, 450), cv2.FONT_HERSHEY_PLAIN, 1, (255, 255, 255), 1, cv2.LINE_AA)

        # img_gray = cv2.cvtColor(im_rd, cv2.COLOR_RGB2GRAY)
        dets = detector(img, 1)
        if len(dets) != 0:
            # 检测到人脸
            for k, d in enumerate(dets):
                # 关键点检测
                shape = sp(img, d)
                # 遍历所有点圈出来
                for pt in shape.parts():
                    pt_pos = (pt.x, pt.y)
                    cv2.circle(img, pt_pos, 2, (0, 255, 0), 1)
                face_descriptor = facerec.compute_face_descriptor(img, shape)
                d_test2 = numpy.array(face_descriptor)
                # 计算欧式距离
                dist = []
                for i in descriptors:
                    dist_ = numpy.linalg.norm(i - d_test2)
                    dist.append(dist_)
                num = dist.index(min(dist))  # 返回最小值

                left_top = (dlib.rectangle.left(d), dlib.rectangle.top(d))
                right_bottom = (dlib.rectangle.right(d), dlib.rectangle.bottom(d))
                cv2.rectangle(img, left_top, right_bottom, (0, 255, 0), 2, cv2.LINE_AA)
                text_point = (dlib.rectangle.left(d), dlib.rectangle.top(d) - 5)
                cv2.putText(img, candidate[num][0:4], text_point, cv2.FONT_HERSHEY_PLAIN, 2.0, (255, 255, 255), 1, 1)  # 标出face

            cv2.putText(img, "facesNum: " + str(len(dets)), (20, 50),  cv2.FONT_HERSHEY_PLAIN, 1.5, (0, 0, 0), 2, cv2.LINE_AA)
        else:
            # 没有检测到人脸
            cv2.putText(img, "facesNum:0", (20, 50),  cv2.FONT_HERSHEY_PLAIN, 1.5, (0, 0, 0), 2, cv2.LINE_AA)
        q_show.put(img)
        # cv2.imshow(name,img)
        # if (cv2.waitKey(1)) == ord('q'):
        #             break


def show():
    # handle for the camera
    # 创建 cv2 摄像头对象
    cv2.namedWindow("out", 1)
    while True:
        frame = q_show.get()
        cv2.imshow('out', frame)
        if (cv2.waitKey(1)) == ord('q'):
            break
    cv2.destroyAllWindows()



if __name__ == '__main__':

    threads = []
    for i in range(5):
        t = threading.Thread(target=output)
        t.start()
        threads.append(t)
    t3 = threading.Thread(target=show)
    t4 = threading.Thread(target=input_q)
    t3.start()
    t4.start()

加速不明显，考虑gpu