口罩检测跟踪

      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()  # 用于停止线程的标识
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/104672051