fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug fix in indexing

Browse Source
This commit is contained in:
David PAGNON 2022-10-25 16:10:45 +02:00 committed by GitHub
parent 3176b8a861
commit fcd5ac0195
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Pose2Sim/triangulate_3d.py
View File

@ -144,7 +144,7 @@ def make_trc(config, Q, keypoints_names, f_range):
Q = zup2yup(Q)
#Add Frame# and Time columns
Q.index = np.array(range(f_range[0]-f_range[0], f_range[1]-f_range[0])) + 1
Q.index = np.array(range(0, f_range[1]-f_range[0])) + 1
Q.insert(0, 't', Q.index / frame_rate)
#Write file
@ -394,7 +394,7 @@ def triangulate_all(config):
json_tracked_files = [[os.path.join(pose_dir, j_dir, j_file) for j_file in json_files_names[j]] for j, j_dir in enumerate(json_dirs_names)]
# Triangulation
f_range = [[0,min([len(j) for j in json_files_names])] if frame_range==[] else [i - min(frame_range) for i in frame_range]][0]
f_range = [[0,min([len(j) for j in json_files_names])] if frame_range==[] else frame_range][0]
n_cams = len(json_dirs_names)
Q_tot, error_tot, nb_cams_excluded_tot = [], [], []
@ -433,11 +433,7 @@ def triangulate_all(config):
Q_tot = Q_tot.apply(interpolate_nans, axis=0, args = [interpolation_kind])
# Create TRC file
trc_path = make_trc(config, Q_tot, keypoints_names, frame_range)
trc_path = make_trc(config, Q_tot, keypoints_names, f_range)
# Recap message
recap_triangulate(config, error_tot, nb_cams_excluded_tot, keypoints_names, trc_path)