first commit

utils/filter.py

@@ -0,0 +1,55 @@
+import cv2
+from  floder.config import cfg
+def filt(video_path):
+    #mask_path = '../../module/ieemoo-ai-search/model/now/ori_old.jpg'
+    maskimg = cv2.imread(cfg.maskImg)
+    fgbg = cv2.createBackgroundSubtractorMOG2(history=500, varThreshold=20, detectShadows = False)
+    capture = cv2.VideoCapture(video_path)
+    ret,frame = capture.read(0)
+    if frame.shape[0]<frame.shape[1]:
+        maskimg = cv2.imread(cfg.maskImg) 
+    else:
+        maskimg = cv2.imread(cfg.maskImg1)
+    print('capture >>>> {}'.format(frame.shape))
+    imglist = []
+    re = False
+    nn = 0
+    while True:
+        ret,frame = capture.read()
+        nn += 1
+        #print('>>>>{}'.format(nn))
+        if not ret:break
+        if not re:
+            re = img_filter(frame, maskimg, fgbg)
+        else:
+            imglist.append(frame)
+            #cv2.imwrite('./imgs/tmpimgs/'+str(nn)+'.jpg', frame)
+        if len(imglist) > 30:
+            break
+    return imglist #-->list imgs
+
+def img_filter(frame, maskimg, fgbg):
+    dic,dics = {},{}
+    iouArea = 0
+
+    height, width = frame.shape[:2]
+    frame = cv2.resize(frame, (int(width/2), int(height/2)), interpolation=cv2.INTER_CUBIC)
+
+    fgmask = fgbg.apply(frame)
+    draw1 = cv2.threshold(fgmask, 25, 255, cv2.THRESH_BINARY)[1]
+
+    draw1 = cv2.bitwise_and(maskimg[:, :, 0], draw1)
+    contours_m, hierarchy_m = cv2.findContours(draw1.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+    for contour in contours_m:
+        dics[len(contour)] = contour
+    if len(dics.keys())>0:
+        cc = sorted(dics.keys())
+        iouArea = cv2.contourArea(dics[cc[-1]])
+    if iouArea>3000 and iouArea<50000:
+        return True
+    return False
+
+if __name__ == '__main__':
+    videoName = 'filterImg.mp4'
+    filt(videoName)
+