diff --git a/__init__.py b/__init__.py index 473bf43..29ab836 100644 --- a/__init__.py +++ b/__init__.py @@ -694,6 +694,7 @@ def execute(self, context): 200), 1) + # TODO add robust joint angle calculation (for various species) """ joint_angles = np.empty(42) joint_angles_conf = np.empty(42) # report confidence on each joint angle @@ -752,7 +753,9 @@ def execute(self, context): pose_output_file = open(bpy.path.abspath(bpy.context.edit_movieclip.filepath)[ :-4] + "_POSE_" + track.name + ".csv", "w") # write header line - pose_output_file.write("frame," + pose_joint_header + ",r1_deg,r2_deg,r3_deg,l1_deg,l2_deg,l3_deg\n") + #pose_output_file.write("frame," + pose_joint_header + ",r1_deg,r2_deg,r3_deg,l1_deg,l2_deg,l3_deg\n") + # TODO add robust joint angle calculation + pose_output_file.write("frame," + pose_joint_header + "\n") for key, value in track_pose.items(): line = key + "," + ",".join(str(e) for e in value.flatten()) pose_output_file.write(line + "\n")