From 44ed2ded3ed1ec730af94b514a577b7460a3cafe Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Fri, 23 Jul 2021 10:26:47 +0300
Subject: [PATCH] [~] Refactor
---
python/tasks/mlb_player.py | 13 ++++++++++++-
1 file changed, 12 insertions(+), 1 deletion(-)
diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py
index ff33dba..36823a1 100644
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@@ -831,6 +831,10 @@ def kernel_7(
plt.imshow(img_canvas[...,::-1])
f.savefig(name)
+ return dict(
+ img_points=img_points,
+ img_canvas=img_canvas,
+ )
# In[5]:
@@ -1002,6 +1006,7 @@ def kernel_14(
return dict(
o_13=o_13,
+ o_7=o_7,
)
def kernel_15(
@@ -1077,7 +1082,10 @@ def kernel_15(
(0, 255, 0),
1,
)
- f = matplotlib.pyplot.figure()
+
+ t32 = o_14['o_7']['estimate_pose'](t13)
+
+ f = matplotlib.pyplot.figure(figsize=(15, 9))
matplotlib.pyplot.title(
'name %s, score %s, frame_id %d' % (
t8['name'],
@@ -1085,7 +1093,10 @@ def kernel_15(
t8.frame_id,
)
)
+ plt.subplot(1, 2, 1)
matplotlib.pyplot.imshow(t14)
+ plt.subplot(1, 2, 2)
+ matplotlib.pyplot.imshow(t32['img_canvas'])
t25 = os.path.join(
t30,
'kernel_15-%s-%05d.jpg' % (