遗留物品检测源码

遗留物品检测算法，可以对着视频直接检测，用到了vibe算法，也做了些改进。

增加功能：过滤红绿灯。

红绿灯：分类判断，或者时间加长

大物体

  

   

    

     

    
    

     

      import random
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      from numba import jit
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     

      from skimage.measure import compare_ssim
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class Boxinfo():
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    def __init__(self,x,y,w,h):
     
    
   

    

     

    
    

     

              self.x=x
     
    
   

    

     

    
    

     

              self.y=y
     
    
   

    

     

    
    

     

              self.w=w
     
    
   

    

     

    
    

     

              self.h=h
     
    
   

    

     

    
    

     

              self.remove_count=0
     
    
   

    

     

    
    

     

      @jit()
     
    
   

    

     

    
    

     

      def initial_background(I_gray, N):
     
    
   

    

     

    
    

     
    # symmetric：对称填充
     
    
   

    

     

    
    

     

          I_pad = np.pad(I_gray, 1, 'symmetric')
     
    
   

    

     

    
    

     

          height = I_pad.shape[0]
     
    
   

    

     

    
    

     

          width = I_pad.shape[1]
     
    
   

    

     

    
    

     

          samples = np.zeros((height, width, N))
     
    
   

    

     

    
    

     
    for i in range(1, height - 1):
     
    
   

    

     

    
    

     
        for j in range(1, width - 1):
     
    
  
 

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

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