From 2292d6ff94a17c4e158c168de23cff0f62603dc1 Mon Sep 17 00:00:00 2001
From: SpeedProg <speedprog@googlemail.com>
Date: Thu, 02 Jan 2020 15:56:20 +0000
Subject: [PATCH] update stuff
---
tsearch.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 48 insertions(+), 0 deletions(-)
diff --git a/tsearch.py b/tsearch.py
new file mode 100644
index 0000000..f29e00c
--- /dev/null
+++ b/tsearch.py
@@ -0,0 +1,48 @@
+#!/bin/python3
+
+import cv2 as cv
+import numpy as np
+from matplotlib import pyplot as plt
+from matplotlib import use, backends
+
+
+use('GTK3Cairo')
+
+if __name__ == '__main__':
+ img = cv.imread('cards_orig.jpg')
+ img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
+ _, img = cv.threshold(img, 50, 255, cv.THRESH_BINARY)
+ img2 = img.copy()
+ #template = cv.imread('data2/icons/m19.png', 0)
+ template = cv.imread('m19_ico3.jpg', 0)
+ template = cv.resize(template, (109, 46), interpolation = cv.INTER_CUBIC)
+ print(template.shape)
+ h, w = template.shape
+ methods = ['cv.TM_CCOEFF', 'cv.TM_CCOEFF_NORMED', 'cv.TM_CCORR',
+ 'cv.TM_CCORR_NORMED', 'cv.TM_SQDIFF', 'cv.TM_SQDIFF_NORMED']
+ for meth in methods:
+ img = img2.copy()
+ method = eval(meth)
+
+ res = cv.matchTemplate(img, template, method)
+ min_val, max_val, min_loc, max_loc = cv.minMaxLoc(res)
+
+ if meth in ['cv.TM_SQDIFF', 'cv.MT_SQDIFF_NORMED']:
+ top_left = min_loc
+ else:
+ top_left = max_loc
+
+ bottom_right = (top_left[0] + w, top_left[1] + h)
+ print(top_left, bottom_right)
+ img2show = cv.cvtColor(img, cv.COLOR_GRAY2BGR)
+ cv.rectangle(img2show, top_left, bottom_right, 1202404, 2)
+ plt.subplot(221), plt.imshow(res, cmap = 'gray')
+ plt.title('Matching Result'), plt.xticks([]), plt.yticks([])
+ plt.subplot(222), plt.imshow(img2show, cmap = 'gray')
+ plt.title('Detected Point'), plt.xticks([]), plt.yticks([])
+ plt.suptitle(meth)
+ plt.subplot(223), plt.imshow(template, cmap = 'gray')
+ plt.title('Template'), plt.xticks([]), plt.yticks([])
+
+ plt.show()
+
--
Gitblit v1.10.0