Worked on card detection

This commit is contained in:
Lukas Wölfer
2019-05-03 00:23:27 +02:00
parent 6c69fdc984
commit fbddffe6c7
3 changed files with 24 additions and 17 deletions

View File

@@ -11,11 +11,12 @@ import cv2
from .context import shenzhen_solitaire
from shenzhen_solitaire.cv import adjustment
from shenzhen_solitaire.cv import card_finder
from shenzhen_solitaire.cv import board_parser
from shenzhen_solitaire import board
from shenzhen_solitaire.cv.configuration import Configuration
def main() -> None:
def generate() -> None:
with open("Solitaire.png", 'rb') as fd:
img_str = fd.read()
nparr = np.frombuffer(img_str, np.uint8)
@@ -24,6 +25,17 @@ def main() -> None:
a = Configuration.generate(image)
a.save('test_config.zip')
def parse() -> board.Board:
with open("Solitaire2.png", 'rb') as fd:
img_str = fd.read()
nparr = np.frombuffer(img_str, np.uint8)
image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
image = cv2.resize(image, (1000, 629))
a = Configuration.load('test_config.zip')
a.field_adjustment = adjustment.adjust_field(image)
return board_parser.parse_board(image, a)
if __name__ == "__main__":
main()
generate()