jens/ocv_template_matching
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix image scaling h,w are swapped

Browse Source
This commit is contained in:
jens
2024-07-01 20:57:16 +02:00
parent 0189e02bbc
commit dcb4215334
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ocv_template_matching.py
+12 -7
View File
@@ -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)