口罩检测跟踪

import os
import time
import datetime

import cv2
import numpy as np
from queue import Queue
from threading import Thread, Event

from onnx_mask.lib.RetinaFace.retinaface import RetinaF
from track.tracker_run import TrackerRun
from video_soure import MyProcess
#from onnx_mask.lib.CenterFace.centerface import CenterFace


class MaskDetectThread(Thread):
    def __init__(self, width, height, videoQueue):
        Thread.__init__(self)
        self.width = width
        self.height = height
        #self.cap = cv2.VideoCapture(0)
        self.videoQueue = videoQueue
        # self.centerface = CenterFace(self.height, self.width)
        self.centerface = RetinaF(self.height, self.width)
        self.MaskNet = cv2.dnn.readNetFromONNX(r'./onnx_mask/model/mobilenetv2.onnx')
        self.__flag = Event()  # 用于暂停线程的标识
        self.__flag.set()  # 设置为True
        self.__running = Event()  # 用于停止线程的标识