CODEWITHSUNDEEP
pythonopencvtemplate-matching
abc123
abc123

Reputation: 1

Template matching with updating templates from previous detection

I am trying to write a code where after matching with a given template, the detected part of that frame becomes the template for the next frame.

temp = "image.png" while True: try: _, frame = cap.read() copy = frame.copy()

    w,h=temp.shape[:-1] 
    res=cv2.matchTemplate(frame,temp,cv2.TM_CCOEFF_NORMED)
    
        
    threshold=0.75
#try:
    
    loc=np.where(res>=threshold)
    print(len(loc))

    for pt in zip(*loc[::-1]):
            
            #cv2.rectangle(img,pt,(pt[0]+w,pt[1]+h),(0,255,255),2)
        point = pt
    

    cropped_image = copy[point[1]:point[1]+h, point[0]:point[0]+w]
    
    temp = cropped_image  #update the template

but after writing this code the template matching is going in totally wrong direction, even though if i remove the "temp = cropped_image" then the cropped_image is actually good.

Upvotes: 0

Views: 162

Answers (1)

Lunethan
Lunethan

Reputation: 76

You can find x,y,w,h of the matched image with cv2.minMaxLoc()

import cv2

src = cv2.imread("source.png", cv2.IMREAD_GRAYSCALE)
templit = cv2.imread("initial_template.png", cv2.IMREAD_GRAYSCALE)

result = cv2.matchTemplate(src, templit, cv2.TM_SQDIFF_NORMED)

minVal, maxVal, minLoc, maxLoc = cv2.minMaxLoc(result)
x, y = minLoc
h, w = templit.shape

cropped_img = src[y: y + h, x: x + w]

//do template matching again with cropped_image

Upvotes: 0

Related Questions