口罩检测跟踪

  

   

    

     

    
    

     

      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