python - オブジェクトを選択して永久に追跡する方法は?

Question

アリを追跡しようとしていますが、アリは形を変えるので非常に難しく、グレースケール ビデオを使用しています。つまり、mosse と camshit は私のプロジェクトでは機能しませんでした。

その位置を取得して分析を行う必要があるため、1 つの特別なもの (ランダムではない) を追跡する必要があります。そのため、1 つのアリを選択してそれを永久に追跡する方法が必要です。すべてを試しましたが、何も機能しませんでした。

まあ、アリの位置を手動で設定することも、マウスで設定することもできます。それは問題ではありません。問題は、最初の位置を設定した後、同じアリを追跡し続ける方法です。

すべてのアリを白く、背景を黒にするため、コードを簡単にするために 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()