From 26116f0b2ed71fc6a8c23fbb6b70f6f0ad5d0369 Mon Sep 17 00:00:00 2001
From: Wlad <9556979+gosticks@users.noreply.github.com>
Date: Tue, 2 Feb 2021 15:26:22 +0100
Subject: [PATCH] render model keypoints to image

---
 example_fit.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/example_fit.py b/example_fit.py
index 276b6b3..20f8feb 100644
--- a/example_fit.py
+++ b/example_fit.py
@@ -91,7 +91,7 @@ r = Renderer()
 init_joints = get_torso(joints)
 init_keypoints = get_torso(keypoints)
 
-print(img_path)
+print("image path:", img_path)
 # setup renderer
 
 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)
 
 # 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(
     init_keypoints,
     radius=0.01,
@@ -154,5 +159,5 @@ train_pose(
     camera=camera,
     renderer=r,
     device=device,
-    iterations=30
+    # iterations=30
 )