目前感觉,利用opencv实现的物体追踪,关键要设置好你所检测对象的area,不然很容易出现混乱。本人也是自学,敬请批评指正。
import cv2
# 定义运算的核算子
BLUR_RADIUS = 21
erode_kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
dilate_kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (9, 9))
# 打开摄像头
cap = cv2.VideoCapture(0)
success, frame = cap.read()
# 丢弃9帧,让相机有足够时间调整
for i in range(9):
success, frame = cap.read()
if not success:
exit(1)
# 取第十帧,并进行模糊操作,作为背景
gray_background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
gray_background = cv2.GaussianBlur(gray_background, (BLUR_RADIUS, BLUR_RADIUS), 0)
# 有了背景的参考图像,开始检测物体,对每一帧转成灰度和高斯模糊
success, frame = cap.read()
frame = cv2.flip(frame, 1)
while success:
gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
gray_frame = cv2.GaussianBlur(gray_frame, (BLUR_RADIUS, BLUR_RADIUS), 0)
# 对两张图片进行差值的绝对值操作
diff = cv2.absdiff(gray_background, gray_frame)
_, thresh = cv2.threshold(diff, 40, 255, cv2.THRESH_BINARY) # 阈值化操作得到黑白图像
# 形态学运算进行平滑处理,便于后续边框的绘制
cv2.erode(thresh, erode_kernel, thresh, iterations=2)
cv2.dilate(thresh, dilate_kernel, thresh, iterations=2)
# 先寻找轮廓
contours, hier = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in contours:
# 排除轮廓太小的物体
if cv2.contourArea(c) > 10000:
x, y, w, h = cv2.boundingRect(c)
cv2.rectangle(frame, (x, y), (x + w, y + h), (255, 255, 0), 2)
cv2.imshow("diff", diff)
cv2.imshow("thresh", thresh)
cv2.imshow("detection", frame)
if cv2.waitKey(1) == 27: # 按下esc键退出
break
success, frame = cap.read()
cv2.flip(frame, 1)
cap.release()
cv2.destroyAllWindows()
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)