From 546a40728f203994f5d65be9031c6db03af16323 Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Tue, 27 Jul 2021 22:47:55 +0300 Subject: [PATCH] [~] Refactor --- python/tasks/mlb_player.py | 90 +++++++++++++++++++++++--------------- 1 file changed, 55 insertions(+), 35 deletions(-) diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py index 1cca897..6dcccb4 100644 --- a/python/tasks/mlb_player.py +++ b/python/tasks/mlb_player.py @@ -1308,32 +1308,8 @@ def kernel_18(o_17): ) def kernel_19(o_18): - import tqdm - import os - import cv2 - import subprocess - - t3 = 'kernel_19-output.dir' - os.makedirs(t3, exist_ok=True) - t6 = [] - for i, o in tqdm.tqdm(enumerate(o_18['t2']['t6'])): - t4 = 'image-%03d.jpg' % i - t5 = os.path.join(t3, t4) - cv2.imwrite(t5, o) - t6.append(t5) - - t7 = os.path.join(t3, 'output.gif') - - subprocess.check_call( - [ - 'convert', - '-delay', - '100', - '-loop', - '0', - *t6, - t7, - ] + kernel_25( + o_18['t2']['t6'][ ) def kernel_20( @@ -1587,15 +1563,6 @@ def kernel_22(o_18): t4=t12, ) -def kernel_23(o_18, o_22, ids=None): - import numpy - - if ids is None: - ids = numpy.s_[:] - - t1 = numpy.arange(len(o_22['t4'])) - t2 = t1[ids] - def kernel_24(img, keypoints): t3 = img.copy() @@ -1616,3 +1583,56 @@ def kernel_24(img, keypoints): return dict( t3=t3, ) + +def kernel_25(images): + import tqdm + import os + import cv2 + import subprocess + + t3 = 'kernel_25-output.dir' + os.makedirs(t3, exist_ok=True) + t6 = [] + for i, o in tqdm.tqdm(enumerate(images)): + t4 = 'image-%03d.jpg' % i + t5 = os.path.join(t3, t4) + cv2.imwrite(t5, o) + t6.append(t5) + + t7 = os.path.join(t3, 'output.gif') + + subprocess.check_call( + [ + 'convert', + '-delay', + '100', + '-loop', + '0', + *t6, + t7, + ] + ) + +def kernel_23(o_18, o_22, ids=None): + import numpy + + if ids is None: + ids = numpy.s_[:] + + t1 = numpy.arange(len(o_22['t4'])) + t2 = t1[ids] + + t7 = [] + for o in t2: + t3 = o_18['t4'][o] + t4 = [ + o['image_canvas'] + for o in o_18['t2']['t1'] + if o['image_name'] == t1[i]['image_id'] + ] + assert len(t4) == 1 + t5 = t4[0] + t6 = kernel_24(t5, t3['keypoints']) + t7.append(t6['t3']) + + kernel_25(t7)