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:03:53 +03:00
parent 1503d7908b
commit 2ac669bcfd
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
d1/mlb_player_v4.py
View File

@ -478,7 +478,12 @@ if use_gpu:
# In[8]:
def estimate_pose(img_ori):
def estimate_pose(img_ori, name=None):
if name is None:
name = tempfile.mktemp(
dir='/kaggle/working',
suffix='.png',
)
# People might be at different scales in the image, perform inference at multiple scales to boost results
scale_param = [0.5, 1.0, 1.5, 2.0]
@ -499,7 +504,7 @@ def estimate_pose(img_ori):
img_canvas = link_key_point(img_points, candidates, subsets)
plt.figure(figsize=(15, 10))
f = plt.figure(figsize=(15, 10))
plt.subplot(1, 2, 1)
plt.imshow(img_points[...,::-1])
@ -507,6 +512,8 @@ def estimate_pose(img_ori):
plt.subplot(1, 2, 2)
plt.imshow(img_canvas[...,::-1])
f.savefig(name)
# In[9]: