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
__init__.py
+3 -2
View File
@@ -235,6 +235,7 @@ class MotionDataVisualizer:
"distance": ("FLOAT", {"default": 7.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"elevation": ("FLOAT", {"default": 120, "min": 0.0, "max": 300.0, "step": 0.1}),
"rotation": ("FLOAT", {"default": -90, "min": -180, "max": 180, "step": 1}),
"poselinewidth": ("FLOAT", {"default": 2, "min": 0, "max": 50, "step": 0.1}),
},
"optional": {
"opt_title": ("STRING", {"default": '' ,"multiline": False}),
@@ -245,12 +246,12 @@ class MotionDataVisualizer:
CATEGORY = "MotionDiff"
FUNCTION = "visualize"
def visualize(self, motion_data, visualization, distance, elevation, rotation, opt_title=None):
def visualize(self, motion_data, visualization, distance, elevation, rotation, poselinewidth, opt_title=None):
pred_motion = motion_data["motion"]
joint = recover_from_ric(pred_motion, 22).numpy()
joint = motion_temporal_filter(joint, sigma=2.5)
pil_frames = plot_3d_motion(
None, t2m_kinematic_chain, joint, distance, elevation, rotation,
None, t2m_kinematic_chain, joint, distance, elevation, rotation, poselinewidth,
title=opt_title if opt_title is not None else '',
fps=1, save_as_pil_lists=True, visualization=visualization
)
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: