如何选择一个对象并永远跟踪它?

2024-06-24 13:34:44 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图跟踪蚂蚁,这真的很难,因为蚂蚁改变了它的形状,我用的是灰度视频。意思是说,莫斯和卡姆希特在我的项目中没有起作用。在

我必须跟踪一个特殊的(不是随机的),因为我需要确定它的位置并进行一些分析。所以我需要一种方法来选择一只蚂蚁并永远跟踪它,我尝试了所有的方法,但是没有任何效果。在

好吧,我可以手动设置蚂蚁的位置,也可以用鼠标,这不是问题,问题是在设置好第一个位置后如何继续跟踪同一只蚂蚁。在

我在我的代码上使用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()

Tags: 方法selfimgifnpcv2kernelcontour