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-27 10:18:23 +03:00
parent 14d5e06aff
commit f02945bd7e
1 changed files with 55 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
python/tasks/mlb_player.py
View File

@ -1336,11 +1336,33 @@ def kernel_19(o_18):
] ]
) )
def kernel_20(o_18): def kernel_20(
o_18,
o_21=None,
):
if o_21 is None:
o_21 = kernel_21()
import cv2 import cv2
import numpy import numpy
import os import os
t1 = numpy.array(o_18['t2']['t7'][0]['keypoints']).reshape(17, -1)
t2 = o_18['t2']['t6'][0]
t3 = o_18['t2']['t1'][0]['image_canvas'].copy()
assert o_18['t2']['t7'][0]['image_id'] == os.path.split(o_18['t2']['t1'][0]['image_name'])[1]
for i, o2 in enumerate(o_21['p_color']):
if i >= 17:
print('fuck %d' % i)
continue
o1 = t1[i, :]
cv2.circle(t3, tuple(o1[:2].astype(numpy.int32)), 3, o2, -1)
cv2.imwrite('output.jpg', cv2.cvtColor(t3, cv2.COLOR_RGB2BGR))
cv2.imwrite('output-v2.jpg', cv2.cvtColor(t2, cv2.COLOR_RGB2BGR))
def kernel_21():
l_pair = [ l_pair = [
(0, 1), (0, 2), (1, 3), (2, 4), # Head (0, 1), (0, 2), (1, 3), (2, 4), # Head
(5, 6), (5, 7), (7, 9), (6, 8), (8, 10), (5, 6), (5, 7), (7, 9), (6, 8), (8, 10),
@ -1386,16 +1408,36 @@ def kernel_20(o_18):
#'Neck', #'Neck',
] ]
t1 = numpy.array(o_18['t2']['t7'][0]['keypoints']).reshape(17, -1) return dict(
t2 = o_18['t2']['t6'][0] l_pair=l_pair,
t3 = o_18['t2']['t1'][0]['image_canvas'].copy() p_color=p_color,
assert o_18['t2']['t7'][0]['image_id'] == os.path.split(o_18['t2']['t1'][0]['image_name'])[1] line_color=line_color,
)
for i, o2 in enumerate(p_color): def kernel_22(o_18):
if i >= 17: t1 = o_18['t2']['t7']
print('fuck %d' % i) t2 = [
continue numpy.array(o['keypoints']).reshape(17, 3)
o1 = t1[i, :] for o in t1
cv2.circle(t3, tuple(o1[:2].astype(numpy.int32)), 3, o2, -1) ]
cv2.imwrite('output.jpg', cv2.cvtColor(t3, cv2.COLOR_RGB2BGR)) t3 = []
cv2.imwrite('output-v2.jpg', cv2.cvtColor(t2, cv2.COLOR_RGB2BGR)) for o in t2:
t4 = numpy.min(o[0])
t5 = numpy.max(o[0])
t6 = numpy.min(o[1])
t7 = numpy.max(o[1])
t8 = (t5 - t4) * (t7 - t6)
t3.append(t8)
t4 = [
dict(
square=t3[i],
keypoints=t2[i],
image_name=o['image_id'],
)
for i, o in enumerate(t1)
]
return dict(
t4=t4,
)