import cv2
import numpy as np

camera = cv2.VideoCapture(0) # 參數(shù)0表示第一個(gè)攝像頭
# 判斷視頻是否打開
if (camera.isOpened()):
  print('Open')
else:
  print('攝像頭未打開')

# 測(cè)試用,查看視頻size
size = (int(camera.get(cv2.CAP_PROP_FRAME_WIDTH)),
    int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT)))
print('size:'+repr(size))

es = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (9, 4))
kernel = np.ones((5, 5), np.uint8)
background = None

while True:
  # 讀取視頻流
  grabbed, frame_lwpCV = camera.read()
  # 對(duì)幀進(jìn)行預(yù)處理，先轉(zhuǎn)灰度圖，再進(jìn)行高斯濾波。
  # 用高斯濾波進(jìn)行模糊處理，進(jìn)行處理的原因：每個(gè)輸入的視頻都會(huì)因自然震動(dòng)、光照變化或者攝像頭本身等原因而產(chǎn)生噪聲。對(duì)噪聲進(jìn)行平滑是為了避免在運(yùn)動(dòng)和跟蹤時(shí)將其檢測(cè)出來。
  gray_lwpCV = cv2.cvtColor(frame_lwpCV, cv2.COLOR_BGR2GRAY)
  gray_lwpCV = cv2.GaussianBlur(gray_lwpCV, (21, 21), 0)

  # 將第一幀設(shè)置為整個(gè)輸入的背景
  if background is None:
    background = gray_lwpCV
    continue
  # 對(duì)于每個(gè)從背景之后讀取的幀都會(huì)計(jì)算其與北京之間的差異，并得到一個(gè)差分圖（different map）。
  # 還需要應(yīng)用閾值來得到一幅黑白圖像，并通過下面代碼來膨脹（dilate）圖像，從而對(duì)孔（hole）和缺陷（imperfection）進(jìn)行歸一化處理
  diff = cv2.absdiff(background, gray_lwpCV)
  diff = cv2.threshold(diff, 25, 255, cv2.THRESH_BINARY)[1] # 二值化閾值處理
  diff = cv2.dilate(diff, es, iterations=2) # 形態(tài)學(xué)膨脹

  # 顯示矩形框
  image, contours, hierarchy = cv2.findContours(diff.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) # 該函數(shù)計(jì)算一幅圖像中目標(biāo)的輪廓
  for c in contours:
    if cv2.contourArea(c) < 1500: # 對(duì)于矩形區(qū)域，只顯示大于給定閾值的輪廓，所以一些微小的變化不會(huì)顯示。對(duì)于光照不變和噪聲低的攝像頭可不設(shè)定輪廓最小尺寸的閾值
      continue
    (x, y, w, h) = cv2.boundingRect(c) # 該函數(shù)計(jì)算矩形的邊界框
    cv2.rectangle(frame_lwpCV, (x, y), (x+w, y+h), (0, 255, 0), 2)

  cv2.imshow('contours', frame_lwpCV)
  cv2.imshow('dis', diff)

  key = cv2.waitKey(1) & 0xFF
  # 按'q'健退出循環(huán)
  if key == ord('q'):
    break
# When everything done, release the capture
camera.release()
cv2.destroyAllWindows()