Worked on additional tools

This commit is contained in:
Lukas Wölfer
2020-06-12 03:58:42 +02:00
parent 9a38c60488
commit 6565792030
6 changed files with 249 additions and 19 deletions

View File

@@ -1,15 +1,42 @@
import argparse
import cv2
import numpy as np
import shenzhen_solitaire.card_detection.configuration as configuration
from shenzhen_solitaire.card_detection import configuration, adjustment, card_finder
from shenzhen_solitaire.card_detection.configuration import Configuration
def main() -> None:
"""Generate a configuration"""
image = cv2.imread("pictures/20190809172213_1.jpg")
parser = argparse.ArgumentParser(
description="Generate pictures for symbols, "
"requires screenshot of field with no moved cards, "
"so 8 columns of 5 cards each"
)
parser.add_argument(
"screenshot_path",
metavar="screenshot_path",
type=str,
help="Path to the screenshot",
)
parser.add_argument(
"--conf",
dest="config_path",
type=str,
default="config.zip",
help="Path to existing config to be merged, or new config",
)
generated_config = configuration.generate(image)
configuration.save(generated_config, "test_config.zip")
args = parser.parse_args()
print(args.screenshot_path)
image = cv2.imread(args.screenshot_path)
adj = adjustment.adjust_field(image)
squares = card_finder.get_field_squares(image, adj, 5, 8)
catalogue = card_finder.catalogue_cards(squares)
generated_config = Configuration(field_adjustment=adj, catalogue=catalogue, meta={})
configuration.save(generated_config, args.config_path)
if __name__ == "__main__":