43 lines
1008 B
Python
43 lines
1008 B
Python
from .context import shenzhen_solitaire
|
|
from shenzhen_solitaire.cv import adjustment
|
|
from shenzhen_solitaire.cv import card_finder
|
|
import cv2
|
|
|
|
A = cv2.imread("Solitaire.png")
|
|
|
|
adj = adjustment.adjust_field(A)
|
|
X = card_finder.get_field_squares(A, adj)
|
|
for h in range(20):
|
|
p = {None: 0}
|
|
for x in X[h]:
|
|
for x2 in ((x1[0], x1[1], x1[2]) for x1 in x):
|
|
if x2 in p:
|
|
p[x2] += 1
|
|
else:
|
|
p[x2] = 1
|
|
B = sorted(p.items(), key=lambda x: x[1])
|
|
print(*B, sep='\n')
|
|
|
|
T = X[h].copy()
|
|
cv2.imshow("Window", T)
|
|
while cv2.waitKey(0) != 27:
|
|
pass
|
|
cv2.destroyWindow("Window")
|
|
assert 0
|
|
|
|
for ix, vx in enumerate(T):
|
|
for iy, vy in enumerate(vx):
|
|
if (vy[0] > 100) and (vy[1] > 100) and (vy[2] > 100):
|
|
T[ix, iy] = [255, 255, 255]
|
|
else:
|
|
T[ix, iy] = [0, 0, 0]
|
|
|
|
cv2.imshow("Window", T)
|
|
cv2.waitKey(0)
|
|
cv2.destroyWindow("Window")
|
|
|
|
|
|
# for j in X:
|
|
# cv2.imshow("Window", j)
|
|
# cv2.waitKey(0)
|