热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

如何使用Python的OpenCV跟踪2个对象的位置?

笼子里有两只老鼠。到目前为止,我已经进行了背景扣除,因此只能看到小鼠。

笼子里有两只老鼠。到目前为止,我已经进行了背景扣除,因此只能看到小鼠。我的最终目标是追踪两只老鼠在笼子中移动时的位置与时间的关系,但我无法弄清楚。我想让两只鼠标的位置出现在屏幕上,也许它们在四处移动时以x-y坐标显示。到目前为止,我已经附加了我的代码。任何帮助将不胜感激!


import cv2
import numpy as np
THRESHOLD_ANIMAL_VS_FLOOR = 70
BGR_COLOR = {'red': (0,255),'green': (127,255,0),'blue': (255,127,'yellow': (0,'black': (0,'white': (255,255)}
cap = cv2.VideoCapture('IMG_0224_Trim3.mp4')
h,w = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)),int(
cap.get(cv2.CAP_PROP_FRAME_WIDTH))
# Check if camera opened successfully
if (cap.isOpened() == False):
print("Error opening video stream or file")
# Read until video is completed
while(cap.isOpened()):
# Capture frame-by-frame
ret,frame = cap.read()
# Convert BGR to GRAY
gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
kernelSize = (5,5)
frameBlur = cv2.GaussianBlur(gray,kernelSize,0)
_,thresh = cv2.threshold(
frameBlur,THRESHOLD_ANIMAL_VS_FLOOR,cv2.THRESH_BINARY)
mask = cv2.inRange(thresh,(0,(180,250,30,0))
contours,hierarchy = cv2.findContours(
thresh.copy(),cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)
for contour in contours:
cv2.drawContours(frame,contour,-1,BGR_COLOR['green'],3,maxLevel=1)
#object tracking ---------------------------------
#-----------------------
cv2.imshow("Frame",frame)
cv2.imshow('mask',mask)
k = cv2.waitKey(5) & 0xFF
if (k == 113): # press q to quit
break
cv2.destroyAllWindows()


您可以编写如下代码:

for contour in contours:
the_size = cv.contourArea(cnt)
M = cv.moments(cnt)
cx = int(M['m10']/M['m00'])
cy = int(M['m01']/M['m00'])
if the_size > A_THRESHOLD:
cv2.puText(frame,"%d,%d"%(cx,cy),(cx,cv2.FONT_HERSHEY_SIMPLEX,1,(0,255,0),cv2.LINE_AA)

推荐阅读
author-avatar
非得高档防水
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有