Minor changes

This commit is contained in:
Lukas Wölfer
2019-05-02 22:44:20 +02:00
parent 0941634776
commit 6c69fdc984
2 changed files with 5 additions and 4 deletions

View File

@@ -22,7 +22,7 @@ class Configuration:
board.Card]],
meta: Dict[str,
str]) -> None:
self.adjustment = adj
self.field_adjustment = adj
self.catalogue = catalogue
self.meta = meta
@@ -33,12 +33,12 @@ class Configuration:
zip_file.writestr(
self.ADJUSTMENT_FILE_NAME, json.dumps(
dataclasses.asdict(
self.adjustment)))
self.field_adjustment)))
counter = 0
for square, card in self.catalogue:
counter += 1
file_stream = io.BytesIO()
np.save(file_stream, square, allow_pickle=False)
np.save(file_stream, card_finder.simplify(square)[0], allow_pickle=False)
file_name = ""
if isinstance(card, board.SpecialCard):
file_name = f's{card.value}-{card.name}-{counter}.npy'

View File

@@ -21,7 +21,8 @@ def main() -> None:
nparr = np.frombuffer(img_str, np.uint8)
image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
Configuration.generate(image)
a = Configuration.generate(image)
a.save('test_config.zip')
if __name__ == "__main__":