Formatting

This commit is contained in:
Lukas Wölfer
2020-02-12 00:53:02 +01:00
parent d752ffb24c
commit a8ead54831
7 changed files with 36 additions and 26 deletions

View File

@@ -219,7 +219,7 @@ def parse_goal(image: np.ndarray, conf: Configuration) -> List[Optional[NumberCa
parse_goal_field(square, conf.catalogue, conf.green_card)
for square in goal_squares
]
return goal_list

View File

@@ -26,6 +26,7 @@ def get_field_squares(
squares.append(get_square(adjustment, index_x, index_y))
return _extract_squares(image, squares)
def catalogue_cards(squares: List[np.ndarray]) -> List[Tuple[np.ndarray, Card]]:
"""Run manual cataloging for given squares"""
cv2.namedWindow("Catalogue", cv2.WINDOW_NORMAL)
@@ -33,7 +34,7 @@ def catalogue_cards(squares: List[np.ndarray]) -> List[Tuple[np.ndarray, Card]]:
result: List[Tuple[np.ndarray, Card]] = []
print("Card ID is [B]ai, [Z]hong, [F]a, [H]ua, [R]ed, [G]reen, [B]lack")
print("Numbercard e.g. R3")
abort_row = 'a'
abort_row = "a"
special_card_map = {
"b": SpecialCard.Bai,
"z": SpecialCard.Zhong,