From dcb4215334ee8e5aa0de409ee6902a6d1dc8e26e Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Mon, 1 Jul 2024 20:57:16 +0200 Subject: [PATCH] bugfix image scaling h,w are swapped --- ocv_template_matching.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/ocv_template_matching.py b/ocv_template_matching.py index 73c0eb9..e5145bc 100644 --- a/ocv_template_matching.py +++ b/ocv_template_matching.py @@ -8,8 +8,8 @@ COLOR_TRACKER = (0, 255, 0) COLOR_TRACKER_EXT = (0, 255, 255) COLOR_MATCHER = (0, 0, 255) COLOR_TEMPLATE = (255, 0, 0) -IMG_SCALE_UP = 1 -TEMPLATE_MATCH_OVERLAP = 1 +IMG_SCALE_UP = 2 +TEMPLATE_MATCH_OVERLAP = 0 video = cv2.VideoCapture("./data/IMG_1730.MP4") @@ -29,7 +29,7 @@ bb_crop = bbox_extend(cv2.selectROI("Image", image, False), TEMPLATE_MATCH_OVERL crop = image_crop(image.copy(), bb_crop) bb_template = cv2.selectROI("Image", crop, False) -template = image_crop(crop, bb_template) +template = image_crop(crop.copy(), bb_template) # Initialize tracker with first frame and bounding box tracker = cv2.TrackerKCF().create() @@ -51,11 +51,16 @@ while True: if ok: # Perform template match operations - crop_scaled = im.resize(crop, IMG_SCALE_UP * crop.shape[0], IMG_SCALE_UP * crop.shape[1]) + crop_scaled = im.resize(crop.copy(), IMG_SCALE_UP * crop.shape[1], IMG_SCALE_UP * crop.shape[0]) template_scaled = im.resize(template, IMG_SCALE_UP * template.shape[0], IMG_SCALE_UP * template.shape[1]) - res = cv2.matchTemplate(crop_scaled, template_scaled, cv2.TM_CCOEFF_NORMED) + res = cv2.matchTemplate(cv2.cvtColor(crop_scaled, cv2.COLOR_BGR2GRAY), cv2.cvtColor(template_scaled, cv2.COLOR_BGR2GRAY), cv2.TM_CCOEFF_NORMED) (min_val, max_val, min_loc, max_loc) = cv2.minMaxLoc(res) - + print(f"(min_val, max_val, min_loc, max_loc): {(min_val, max_val, min_loc, max_loc)}") + print(f"matcher_res : {res.shape}") + print(f"crop : {crop.shape}") + print(f"crop_scaled : {crop_scaled.shape}") + print(f"template : {template.shape}") + print(f"template_scaled : {template_scaled.shape}") matcher_bbox_local = (max_loc[0]/IMG_SCALE_UP, max_loc[1]/IMG_SCALE_UP, template.shape[1], template.shape[0]) matcher_top_left_local, matcher_bottom_right_local = to_rect(bbox_round(matcher_bbox_local)) @@ -65,7 +70,7 @@ while True: tracker_rect = to_rect(bb_crop) template_top_left, template_bottom_right = to_rect(bb_template) - cv2.rectangle(crop, template_top_left, template_bottom_right, COLOR_TEMPLATE, 3) + cv2.rectangle(crop, template_top_left, template_bottom_right, COLOR_TEMPLATE, 1) cv2.rectangle(crop, matcher_top_left_local, matcher_bottom_right_local, COLOR_MATCHER, 1) cv2.rectangle(image_anno, matcher_rect[0], matcher_rect[1], COLOR_MATCHER, 1)