This commit is contained in:
Lukas Wölfer
2020-02-12 23:45:51 +01:00
parent 7247f465de
commit ca44b8977d
9 changed files with 167 additions and 74 deletions

41
benchmark/unsolved.py Normal file
View File

@@ -0,0 +1,41 @@
import multiprocessing
import tempfile
import time
from pathlib import Path
import cv2
import numpy as np
import shenzhen_solitaire.card_detection.configuration as configuration
import shenzhen_solitaire.solver.solver as solver
from shenzhen_solitaire.card_detection.board_parser import parse_board
from .util import run_benchmark
benchmark_files = [
"pictures/unsolved/tmp1ern14si.png",
"pictures/unsolved/tmp2_0vn4tl.png",
"pictures/unsolved/tmp32jmcnfp.png",
"pictures/unsolved/tmpcml2ldfl.png",
"pictures/unsolved/tmpd7rbwwdb.png",
"pictures/unsolved/tmpdudxuw0s.png",
"pictures/unsolved/tmpeplvz9bk.png",
"pictures/unsolved/tmph_esy__3.png",
"pictures/unsolved/tmpn95ueb7_.png",
"pictures/unsolved/tmpqzay4q08.png",
"pictures/unsolved/tmputbych59.png",
"pictures/unsolved/tmpx4uo6pg3.png",
]
def main() -> None:
with multiprocessing.Pool() as pool:
result = pool.imap_unordered(
run_benchmark, [Path(benchmark) for benchmark in benchmark_files]
)
for current_result in result:
print(current_result)
if __name__ == "__main__":
main()