我试图跟踪蚂蚁,这真的很难,因为蚂蚁改变了它的形状,我用的是灰度视频。意思是说,莫斯和卡姆希特在我的项目中没有起作用。在
我必须跟踪一个特殊的(不是随机的),因为我需要确定它的位置并进行一些分析。所以我需要一种方法来选择一只蚂蚁并永远跟踪它,我尝试了所有的方法,但是没有任何效果。在
好吧,我可以手动设置蚂蚁的位置,也可以用鼠标,这不是问题,问题是在设置好第一个位置后如何继续跟踪同一只蚂蚁。在
我在我的代码上使用knn过滤器来简化它,因为我会让我的蚂蚁都是白色的,背景是黑色的。在
这是我的准则:
import numpy as np
import cv2
class ColourTracker:
def __init__(self):
cv2.namedWindow("Background")
cv2.namedWindow("Frame")
self.capture = cv2.VideoCapture('video.avi')
self.knn = cv2.createBackgroundSubtractorKNN()
def run(self):
while True:
f, orig_img = self.capture.read()
if( not f ):
break;
fore = self.knn.apply( orig_img )
back = self.knn.getBackgroundImage()
kernel = np.ones((5,5),np.uint8)
fore = cv2.erode( fore, kernel )
fore = cv2.dilate( fore, kernel )
# fore = fore[r:r+h, c:c+w]
image, contours, hiearchy = cv2.findContours( fore, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE )
maximumArea = 0
bestContour = None
for contour in contours:
currentArea = cv2.contourArea(contour)
if currentArea > maximumArea:
bestContour = contour
maximumArea = currentArea
x,y,w,h = cv2.boundingRect(bestContour)
cv2.rectangle(orig_img, (x,y),(x+w,y+h), (0,0,255), 3)
cv2.imshow( "Background", back )
cv2.imshow( "Frame", orig_img )
k = cv2.waitKey(24) & 0xff
print k
if k == 27:
break
if __name__ == "__main__":
colour_tracker = ColourTracker()
colour_tracker.run()
目前没有回答
相关问题 更多 >
编程相关推荐