Worked on detecting card border

This commit is contained in:
Lukas Wölfer
2020-02-06 21:42:18 +01:00
parent cff356c6c4
commit cf89e4c694
7 changed files with 237 additions and 181 deletions

23
tools/generate_border.py Normal file
View File

@@ -0,0 +1,23 @@
import numpy as np
import cv2
from shenzhen_solitaire.card_detection.configuration import Configuration
import shenzhen_solitaire.card_detection.adjustment as adjustment
import shenzhen_solitaire.card_detection.card_finder as card_finder
import copy
def main() -> None:
"""Generate a configuration"""
image = cv2.imread("pictures/20190809172213_1.jpg")
border_adjustment = adjustment.adjust_squares(image, count_x=8, count_y=13)
border_square_pos = adjustment.adjust_squares(
image, count_x=1, count_y=1, adjustment=copy.deepcopy(border_adjustment)
)
border_square = card_finder.get_field_squares(image, border_square_pos, 1, 1)
empty_square = card_finder.get_field_squares(image, border_square_pos, 1, 1)
if __name__ == "__main__":
main()

View File

@@ -1,14 +1,16 @@
import numpy as np
import cv2
from shenzhen_solitaire.card_detection.configuration import Configuration
import shenzhen_solitaire.card_detection.configuration as configuration
def main() -> None:
"""Generate a configuration"""
image = cv2.imread("pictures/20190809172213_1.jpg")
generated_config = Configuration.generate(image)
generated_config.save('test_config.zip')
generated_config = configuration.generate(image)
configuration.save(generated_config, "test_config.zip")
if __name__ == "__main__":
main()