vhad/body-pose-animation
1
0
Fork 0
mirror of https://github.com/gosticks/body-pose-animation.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

render model keypoints to image

Browse Source
This commit is contained in:
Wlad 2021-02-02 15:26:22 +01:00
parent acbd831059
commit 26116f0b2e
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
example_fit.py
View File

@ -91,7 +91,7 @@ r = Renderer()
init_joints = get_torso(joints) init_joints = get_torso(joints)
init_keypoints = get_torso(keypoints) init_keypoints = get_torso(keypoints)
print(img_path) print("image path:", img_path)
# setup renderer # setup renderer
r.render_model(model, model_out) r.render_model(model, model_out)
@ -100,6 +100,11 @@ r.render_model(model, model_out)
r.render_image_from_path(img_path, est_scale) r.render_image_from_path(img_path, est_scale)
# render openpose torso markers # render openpose torso markers
render_keypoints = r.render_points(
keypoints,
radius=0.005,
color=[1.0, 1.0, 1.0, 1.0])
render_keypoints = r.render_keypoints( render_keypoints = r.render_keypoints(
init_keypoints, init_keypoints,
radius=0.01, radius=0.01,
@ -154,5 +159,5 @@ train_pose(
camera=camera, camera=camera,
renderer=r, renderer=r,
device=device, device=device,
iterations=30 # iterations=30
) )