Delete viz_utils.py

viz_utils.py

@@ -1,50 +0,0 @@
-import numpy as np
-import plotly.graph_objects as go
-
-CHAIN = [-1, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 9, 9, 12, 13, 14, 16, 17, 18, 19]
-
-JOINT_NAMES = [
-    "Hips", "RightUpLeg", "RightLeg", "RightFoot", "LeftUpLeg", "LeftLeg", "LeftFoot",
-    "Spine", "Spine1", "Spine2", "Neck", "Head", "RightShoulder", "RightArm", "RightForeArm",
-    "RightHand", "LeftShoulder", "LeftArm", "LeftForeArm", "LeftHand", "RightToeBase", "LeftToeBase"
-]
-
-def save_bvh_for_blender(xyz_data, filename):
-    seq_len, n_joints, _ = xyz_data.shape
-    with open(filename, 'w') as f:
-        f.write("HIERARCHY\n")
-        def write_joint(idx, indent_level):
-            indent = "\t" * indent_level
-            name = JOINT_NAMES[idx] if idx < len(JOINT_NAMES) else f"Joint_{idx}"
-            if idx == 0: f.write(f"{indent}ROOT {name}\n")
-            else: f.write(f"{indent}JOINT {name}\n")
-            f.write(f"{indent}{{\n")
-            f.write(f"{indent}\tOFFSET 0.00 0.00 0.00\n") 
-            f.write(f"{indent}\tCHANNELS 3 Xposition Yposition Zposition\n")
-            children = [i for i, p in enumerate(CHAIN) if p == idx]
-            if len(children) > 0:
-                for child in children: write_joint(child, indent_level + 1)
-            else:
-                f.write(f"{indent}\tEnd Site\n{indent}\t{{\n{indent}\t\tOFFSET 0.00 0.00 0.00\n{indent}\t}}\n")
-            f.write(f"{indent}}}\n")
-        write_joint(0, 0)
-        f.write(f"MOTION\nFrames: {seq_len}\nFrame Time: 0.050000\n")
-        for i in range(seq_len):
-            row = xyz_data[i].flatten()
-            f.write(" ".join([f"{val:.6f}" for val in row]) + "\n")
-    return filename
-
-def plot_skeleton(xyz_data):
-    xyz = xyz_data[::4] 
-    xyz = xyz - xyz[0:1, 0:1, :]
-    frames = []
-    for k in range(len(xyz)):
-        x_p, y_p, z_p = [], [], []
-        for child, parent in enumerate(CHAIN):
-            if parent == -1 or child >= xyz.shape[1]: continue
-            x_p.extend([xyz[k, parent, 0], xyz[k, child, 0], None])
-            y_p.extend([xyz[k, parent, 2], xyz[k, child, 2], None]) 
-            z_p.extend([xyz[k, parent, 1], xyz[k, child, 1], None])
-        frames.append(go.Frame(data=[go.Scatter3d(x=x_p, y=y_p, z=z_p, mode='lines+markers', marker=dict(size=4, color='red'), line=dict(color='blue', width=4))]))
-    fig = go.Figure(data=frames[0].data, layout=go.Layout(title="Motion Preview", scene=dict(xaxis=dict(range=[-2, 2], autorange=False), yaxis=dict(range=[-2, 2], autorange=False), zaxis=dict(range=[-0, 2], autorange=False), aspectmode='cube'), updatemenus=[dict(type="buttons", buttons=[dict(label="Play", method="animate", args=[None])])]), frames=frames)
-    return fig