lukas/shenzhen-solitaire
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor changes

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

6
shenzhen_solitaire/cv/configuration.py
View File

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

3
test/cv_helper.py
View File

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