import cv2
cap = cv2.VideoCapture("traffic.mp4")
backSub = cv2.createBackgroundSubtractorMOG2(
history=500, varThreshold=50, detectShadows=True
)
while True:
ok, frame = cap.read()
if not ok:
break
fg = backSub.apply(frame)
cv2.imshow("fg", fg)
if cv2.waitKey(30) & 0xFF == 27:
break
cap.release()
cv2.destroyAllWindows()
Meanshift 目标跟踪
基于直方图反投影的区域跟踪
Pythonimport cv2
cap = cv2.VideoCapture(0)
ok, frame = cap.read()
x, y, w, h = 100, 100, 200, 200 # 初始 ROI
track_window = (x, y, w, h)
roi = frame[y:y+h, x:x+w]
hsv = cv2.cvtColor(roi, cv2.COLOR_BGR2HSV)
mask = cv2.inRange(hsv, (0, 60, 32), (180, 255, 255))
roi_hist = cv2.calcHist([hsv], [0], mask, [180], [0, 180])
cv2.normalize(roi_hist, roi_hist, 0, 255, cv2.NORM_MINMAX)
term_crit = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1)
while True:
ok, frame = cap.read()
if not ok:
break
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
dst = cv2.calcBackProject([hsv], [0], roi_hist, [0, 180], 1)
ret, track_window = cv2.meanShift(dst, track_window, term_crit)
x, y, w, h = track_window
cv2.rectangle(frame, (x, y), (x+w, y+h), 255, 2)
cv2.imshow("track", frame)
if cv2.waitKey(30) & 0xFF == 27:
break