✔️ 移动对象分析,我们可以绘制对象运行轨迹曲线,这个主要是根据移动对象窗口轮廓,获取中心位置,然后使用中心位置进行绘制即可得到。
✔️ 大致的程序步骤如下:
- 初始化路径点数组
- 对每帧的预测轮廓提取中心位置添加到路径数组
- 绘制路径曲线
✔️ 我们只需要在上一节Camshift的代码中添加一个数组存储中心位置,然后使用cv2.line()
画出即可。
修改部分:
tracking_path = []
while True:
ret, frame = cap.read()
if ret is False:
break;
hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV)
dst = cv.calcBackProject([hsv],[0],roi_hist,[0,180],1)
# ,搜索更新roi区域, 保持运行轨迹
ret, track_box = cv.CamShift(dst, track_window, term_crit)
track_window = track_box
# 椭圆中心
pt = np.int32(ret[0])
if pt[0] > 0 and pt[1] > 0:
tracking_path.append(pt)
# 绘制跟踪对象位置窗口与对象运行轨迹
#cv.ellipse(frame, ret, (0, 0, 255), 3, 8)
for i in range(1, len(tracking_path)):
cv.line(frame, (tracking_path[i - 1][0], tracking_path[i - 1][1]),
(tracking_path[i][0], tracking_path[i][1]), (0, 255, 0), 2, 6, 0)
cv.imshow('Demo',frame)
k = cv.waitKey(50) & 0xff
if k == 27:
break
else:
cv.imwrite(chr(k)+".jpg",frame)