66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
import cv2
|
||
import numpy as np
|
||
|
||
base = 'c:/Users/27942/Desktop/codes/codex_jxs_code/images/'
|
||
main_img = cv2.imread(base + '1.jpg')
|
||
main_gray = cv2.cvtColor(main_img, cv2.COLOR_BGR2GRAY)
|
||
|
||
def match_template_alpha(main_gray, alpha, name):
|
||
h, w = alpha.shape[:2]
|
||
best = None
|
||
|
||
# 策略1: 直接用alpha通道匹配灰度图
|
||
# 策略2: 用alpha做mask，只比较非透明区域
|
||
# 策略3: 对alpha二值化后匹配
|
||
|
||
_, alpha_bin = cv2.threshold(alpha, 10, 255, cv2.THRESH_BINARY)
|
||
|
||
scales = [1.0] + list(np.linspace(0.65, 1.35, 29))
|
||
for scale in scales:
|
||
rw, rh = int(w * scale), int(h * scale)
|
||
if rh < 10 or rw < 10:
|
||
continue
|
||
if rh > main_gray.shape[0] or rw > main_gray.shape[1]:
|
||
continue
|
||
|
||
resized_alpha = cv2.resize(alpha, (rw, rh))
|
||
resized_bin = cv2.resize(alpha_bin, (rw, rh))
|
||
_, resized_mask = cv2.threshold(resized_bin, 10, 255, cv2.THRESH_BINARY)
|
||
|
||
# 用mask匹配（只看非透明区域）
|
||
res = cv2.matchTemplate(main_gray, resized_alpha, cv2.TM_CCOEFF_NORMED, mask=resized_mask)
|
||
_, max_val, _, max_loc = cv2.minMaxLoc(res)
|
||
|
||
if best is None or max_val > best['conf']:
|
||
best = {
|
||
'x': max_loc[0], 'y': max_loc[1],
|
||
'w': rw, 'h': rh,
|
||
'conf': max_val, 'scale': scale,
|
||
'cx': max_loc[0] + rw // 2,
|
||
'cy': max_loc[1] + rh // 2,
|
||
}
|
||
|
||
return best
|
||
|
||
results = {}
|
||
for name in ['2.png', '3.png', '4.png']:
|
||
template = cv2.imread(base + name, cv2.IMREAD_UNCHANGED)
|
||
alpha = template[:, :, 3]
|
||
best = match_template_alpha(main_gray, alpha, name)
|
||
results[name] = best
|
||
x, y, cx, cy = best['x'], best['y'], best['cx'], best['cy']
|
||
conf, scale = best['conf'], best['scale']
|
||
print(name + ': pos=(' + str(x) + ',' + str(y) + ') center=(' + str(cx) + ',' + str(cy) + ') conf=' + str(round(conf, 4)) + ' scale=' + str(round(scale, 2)))
|
||
|
||
# 可视化
|
||
colors = [(0, 0, 255), (0, 255, 0), (255, 0, 0)]
|
||
vis = main_img.copy()
|
||
for i, name in enumerate(['2.png', '3.png', '4.png']):
|
||
m = results[name]
|
||
cv2.rectangle(vis, (m['x'], m['y']), (m['x'] + m['w'], m['y'] + m['h']), colors[i], 2)
|
||
cv2.putText(vis, name, (m['x'], m['y'] - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, colors[i], 1)
|
||
cv2.circle(vis, (m['cx'], m['cy']), 5, colors[i], -1)
|
||
|
||
cv2.imwrite(base + 'result.jpg', vis)
|
||
print('result saved to images/result.jpg')
|