Pownie/advancedskrald
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Gonna catch em all.

Browse Source
This commit is contained in:
Casper 2019-04-24 09:41:57 +02:00
parent f2a2534c34
commit cde518c14e
1 changed files with 20 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
adapter.py
View File

@ -7,10 +7,11 @@ import numpy as np
from main import find_occupied_squares from main import find_occupied_squares
from runner import find_homography, warp_board from runner import find_homography, warp_board
# Load base64 encoded image from stdin
from tensor_classifier import predict_board from tensor_classifier import predict_board
while True: while True:
try:
stdin = sys.stdin.readline() stdin = sys.stdin.readline()
stdin_decoded = base64.b64decode(stdin) stdin_decoded = base64.b64decode(stdin)
img_array = np.frombuffer(stdin_decoded, dtype=np.uint8) img_array = np.frombuffer(stdin_decoded, dtype=np.uint8)
@ -30,3 +31,5 @@ while True:
} }
print(json.dumps(result)) print(json.dumps(result))
except Exception as e:
print(e)