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

Renamed project.

Browse Source
This commit is contained in:
Daniel
2024-10-14 16:28:34 +02:00
parent 256fd26f3f
commit 9e03533704
15 changed files with 44 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/test_skelda_dataset.py
View File

@ -12,8 +12,8 @@ import test_triangulate
import utils_2d_pose
from skelda import evals, utils_pose
sys.path.append("/SimplePoseTriangulation/swig/")
import spt
sys.path.append("/RapidPoseTriangulation/swig/")
import rpt
# ==================================================================================================
@ -100,7 +100,7 @@ if dataset_use.endswith("_wb"):
# eval_joints[eval_joints.index("head")] = "nose"
eval_joints = list(joint_names_2d)
# output_dir = "/SimplePoseTriangulation/data/testoutput/"
# output_dir = "/RapidPoseTriangulation/data/testoutput/"
output_dir = ""
# ==================================================================================================
@ -271,7 +271,7 @@ def main():
all_ids = []
all_paths = []
times = []
triangulator = spt.Triangulator(min_score=minscore, min_group_size=min_group_size)
triangulator = rpt.Triangulator(min_score=minscore, min_group_size=min_group_size)
old_scene = ""
old_index = -1
for label in tqdm.tqdm(labels):
@ -318,10 +318,10 @@ def main():
if sum(np.sum(p) for p in poses_2d) == 0:
poses3D = np.zeros([1, len(joint_names_3d), 4]).tolist()
else:
spt_cameras = spt.convert_cameras(label["cameras"])
rpt_cameras = rpt.convert_cameras(label["cameras"])
roomparams = [label["room_size"], label["room_center"]]
poses3D = triangulator.triangulate_poses(
poses_2d, spt_cameras, roomparams, joint_names_2d
poses_2d, rpt_cameras, roomparams, joint_names_2d
)
time_3d = time.time() - start