yatman
/
Image_Gen_Server
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update openpose_gen

This commit is contained in:
zaqxs123456 2024-10-07 14:02:19 +08:00
parent f6bb2ef081
commit 353e7c1566
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
openpose_gen.py
View File

@ -93,7 +93,7 @@ def save_bodypose(width: int, height: int, coordinates: list):
canvas = draw_bodypose(canvas, keypoints, skel.coco_limbSeq, skel.coco_colors)
# Save as body_pose_output0000.png, body_pose_output0001.png, ...
cv2.imwrite('body_pose_output%04d.png' % save_bodypose.counter, canvas)
cv2.imwrite('output/body_pose_output%04d.png' % save_bodypose.counter, canvas)
save_bodypose.counter += 1 # Increment the counter
def main():