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

Add SMPL2OpenPose

Browse Source
This commit is contained in:
Hacker 17082006
2024-04-01 23:09:18 +07:00
parent d23aaf44e1
commit 3f26363748
3 changed files with 52 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
motiondiff_modules/hmr2/utils/render_openpose.py
+4 -5
View File
@@ -52,8 +52,7 @@ def render_keypoints(img: np.array,
Returns:
(np.array): Image of shape (H, W, 3) with keypoints drawn on top of the original image.
"""
img_orig = img.copy()
width, height = img.shape[1], img.shape[2]
width, height = img.shape[1], img.shape[0]
area = width * height
lineType = 8
@@ -76,8 +75,8 @@ def render_keypoints(img: np.array,
thicknessLineScaled = int(round(min(thicknessLine[index1], thicknessLine[index2]) * pose_scales[0]))
colorIndex = index2
color = colors[colorIndex % numberColors]
keypoint1 = keypoints[index1, :-1].astype(np.int)
keypoint2 = keypoints[index2, :-1].astype(np.int)
keypoint1 = keypoints[index1, :-1].astype(np.int32)
keypoint2 = keypoints[index2, :-1].astype(np.int32)
cv2.line(img, tuple(keypoint1.tolist()), tuple(keypoint2.tolist()), tuple(color.tolist()), thicknessLineScaled, lineType, shift)
for part in range(len(keypoints)):
faceIndex = part
@@ -86,7 +85,7 @@ def render_keypoints(img: np.array,
thicknessCircleScaled = int(round(thicknessCircle[faceIndex] * pose_scales[0]))
colorIndex = part
color = colors[colorIndex % numberColors]
center = keypoints[faceIndex, :-1].astype(np.int)
center = keypoints[faceIndex, :-1].astype(np.int32)
cv2.circle(img, tuple(center.tolist()), radiusScaled, tuple(color.tolist()), thicknessCircleScaled, lineType, shift)
return img