Made image detection work
This commit is contained in:
@@ -8,17 +8,18 @@ import numpy as np
|
||||
from shenzhen_solitaire import board
|
||||
from shenzhen_solitaire.card_detection import adjustment, board_parser
|
||||
from shenzhen_solitaire.card_detection.configuration import Configuration
|
||||
from . import boards
|
||||
|
||||
|
||||
class CardDetectionTest(unittest.TestCase):
|
||||
def test_parse(self) -> None:
|
||||
"""Parse a configuration"""
|
||||
with open("pictures/20190809172213_1.jpg", "rb") as png_file:
|
||||
img_str = png_file.read()
|
||||
nparr = np.frombuffer(img_str, np.uint8)
|
||||
image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
|
||||
# image = cv2.resize(image, (1000, 629))
|
||||
"""Parse a configuration and a board"""
|
||||
image = cv2.imread("pictures/20190809172206_1.jpg")
|
||||
|
||||
loaded_config = Configuration.load("test_config.zip")
|
||||
# loaded_config.field_adjustment = adjustment.adjust_field(image)
|
||||
print(board_parser.parse_board(image, loaded_config))
|
||||
my_board = board_parser.parse_board(image, loaded_config)
|
||||
|
||||
for rows in zip(boards.B20190809172206_1.field, my_board.field):
|
||||
for good_cell, test_cell in zip(*rows):
|
||||
self.assertEqual(good_cell, test_cell)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user