From 0486675c1c0ee188faa553e9da444a932a2e3928 Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Tue, 20 Jul 2021 10:05:11 +0300 Subject: [PATCH] [~] Refactor --- python/tasks/mlb_player.py | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py index 4e371cf..4e0b20e 100644 --- a/python/tasks/mlb_player.py +++ b/python/tasks/mlb_player.py @@ -224,13 +224,24 @@ def kernel_5(o_4): ] ) +def kernel_12(): + import easyocr + t6 = easyocr.Reader(['en']) + + return dict( + t6=t6, + ) + def kernel_6( o_7=None, o_10=None, + o_12=None, + max_frames=None, ): - import easyocr + if max_frames is None: + max_frames = 10 + import cv2 - t6 = easyocr.Reader(['en']) t1 = glob.glob('*.mp4') @@ -244,7 +255,11 @@ def kernel_6( t3 = t2.read() assert t3[0] t4 = t3[1] - t5 = t6.readtext(t4) + if not o_12 is None: + t5 = o_12['t6'].readtext(t4) + else: + t5 = None + if not o_7 is None: t10 = o_7['estimate_pose'](t4) else: