fxreader.online/freelance-project-34-marketing-blog
2
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

[~] Refactor

Browse Source
This commit is contained in:
Siarhei Siniak 2021-07-20 08:31:20 +03:00
parent d10c5664f5
commit 6e786e3de0
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
python/tasks/mlb_player.py
View File

@ -756,7 +756,11 @@ def kernel_7(
img_ori, img_ori,
name=None, name=None,
scale_param=None, scale_param=None,
display=None,
): ):
if display is None:
display = True
if scale_param is None: if scale_param is None:
scale_param = [0.5, 1.0, 1.5, 2.0] scale_param = [0.5, 1.0, 1.5, 2.0]
@ -784,18 +788,19 @@ def kernel_7(
subsets, img_points = draw_key_point(subsets, peaks, img_ori) subsets, img_points = draw_key_point(subsets, peaks, img_ori)
# After predicting Heatmaps and PAF's, proceeed to link joints correctly # After predicting Heatmaps and PAF's, proceeed to link joints correctly
img_canvas = link_key_point(img_points, candidates, subsets) if display:
img_canvas = link_key_point(img_points, candidates, subsets)
f = plt.figure(figsize=(15, 10)) f = plt.figure(figsize=(15, 10))
plt.subplot(1, 2, 1) plt.subplot(1, 2, 1)
plt.imshow(img_points[...,::-1]) plt.imshow(img_points[...,::-1])
plt.subplot(1, 2, 2) plt.subplot(1, 2, 2)
plt.imshow(img_canvas[...,::-1]) plt.imshow(img_canvas[...,::-1])
f.savefig(name) f.savefig(name)
# In[5]: # In[5]:
@ -821,7 +826,7 @@ def kernel_8(
def kernel_9_benchmark( def kernel_9_benchmark(
o_7, o_7,
): ):
t1 = o_7['cv2'].imread('../input/indonesian-traditional-dance/tgagrakanyar/tga_0000.jpg' t1 = o_7['cv2'].imread('../input/indonesian-traditional-dance/tgagrakanyar/tga_0000.jpg')
t5 = 10 t5 = 10
t2 = datetime.datetime.now() t2 = datetime.datetime.now()
for k in range(t5): for k in range(t5):