nodes/Fannovel16-ComfyUI-MotionDiff
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add linewidth as variable

Browse Source 
Thicker maybe better for openpose cn
This commit is contained in:
kijai
2023-10-16 18:46:46 +03:00
parent a1e2df0d7b
commit d3ded2a328
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mogen/utils/plot_utils.py
+2 -2
View File
@@ -142,7 +142,7 @@ def recover_from_ric(data, joints_num):
return positions
def plot_3d_motion(save_path, kinematic_tree, joints, distance, elevation, rotation, title, figsize=(10, 10), fps=120, radius=4, visualization="original", save_as_pil_lists=False):
def plot_3d_motion(save_path, kinematic_tree, joints, distance, elevation, rotation, poselinewidth, title, figsize=(10, 10), fps=120, radius=4, visualization="original", save_as_pil_lists=False):
matplotlib.use('Agg')
title_sp = title.split(' ')
@@ -226,7 +226,7 @@ def plot_3d_motion(save_path, kinematic_tree, joints, distance, elevation, rotat
linewidth = 2.0
if visualization == "pseudo-openpose":
for j in range(len(data[index, chain, 0]) - 1):
ax.plot3D(data[index, chain, 0][j:j+2], data[index, chain, 1][j:j+2], data[index, chain, 2][j:j+2], linewidth=linewidth,
ax.plot3D(data[index, chain, 0][j:j+2], data[index, chain, 1][j:j+2], data[index, chain, 2][j:j+2], linewidth=poselinewidth,
color=color[j % len(color)])
else: