crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated skelda version.

Browse Source
This commit is contained in:
Daniel
2024-10-14 16:10:19 +02:00
parent 87a205c36c
commit 256fd26f3f
4 changed files with 10 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
scripts/test_triangulate.py
View File

@ -9,9 +9,8 @@ import cv2
import matplotlib
import numpy as np
import draw_utils
import utils_2d_pose
from skelda import utils_pose
from skelda import utils_pose, utils_view
sys.path.append("/SimplePoseTriangulation/swig/")
import spt
@ -324,14 +323,14 @@ def main():
print("2D time:", time.time() - stime)
# print([np.array(p).round(6).tolist() for p in poses_2d])
fig1 = draw_utils.show_poses2d(
poses_2d, np.array(images_2d), joint_names_2d, "2D detections"
fig1 = utils_view.draw_many_images(
sample["imgpaths_color"], [], [], poses_2d, joint_names_2d, "2D detections"
)
fig1.savefig(os.path.join(dirpath, "2d-k.png"), dpi=fig1.dpi)
# draw_utils.utils_view.show_plots()
if len(images_2d) == 1:
draw_utils.utils_view.show_plots()
utils_view.show_plots()
continue
# Get 3D poses
@ -361,15 +360,13 @@ def main():
# print(poses2D)
# print(poses3D.round(3).tolist())
fig2 = draw_utils.utils_view.show_poses3d(
poses3D, joint_names_3d, roomparams, camparams
)
fig3 = draw_utils.show_poses2d(
poses2D, np.array(images_2d), joint_names_3d, "2D reprojections"
fig2 = utils_view.draw_poses3d(poses3D, joint_names_3d, roomparams, camparams)
fig3 = utils_view.draw_many_images(
sample["imgpaths_color"], [], [], poses2D, joint_names_3d, "2D projections"
)
fig2.savefig(os.path.join(dirpath, "3d-p.png"), dpi=fig2.dpi)
fig3.savefig(os.path.join(dirpath, "2d-p.png"), dpi=fig3.dpi)
draw_utils.utils_view.show_plots()
utils_view.show_plots()
# ==================================================================================================