From 01a1173e13baef50ebed85d8c295b2eeb9fd3476 Mon Sep 17 00:00:00 2001 From: zchengrong <124802516+zchengrong@users.noreply.github.com> Date: Mon, 7 Apr 2025 17:52:16 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=20=20pose=20transform=20=E9=83=A8=E7=BD=B2=20fix?= =?UTF-8?q?=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20docs=EF=BC=88?= =?UTF-8?q?=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refactor?= =?UTF-8?q?=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E6=B5=8B=E8=AF=95):?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../generate_image/utils/pose_transform_upload.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/app/service/generate_image/utils/pose_transform_upload.py b/app/service/generate_image/utils/pose_transform_upload.py index d768ffa..0310abb 100644 --- a/app/service/generate_image/utils/pose_transform_upload.py +++ b/app/service/generate_image/utils/pose_transform_upload.py @@ -52,9 +52,10 @@ def upload_video(frames, user_id, category, file_name): try: object_name = f'{user_id}/{category}/{file_name}' # 创建视频写入器 - fps = 24 # 帧率 - fourcc = cv2.VideoWriter_fourcc(*'mp4v') - video_writer = cv2.VideoWriter(file_name, fourcc, fps, (768, 512)) + fps = 1 # 帧率 + + fourcc = cv2.VideoWriter.fourcc(*'mp4v') + video_writer = cv2.VideoWriter(file_name, fourcc, fps, (512, 768)) # 逐帧写入 for frame in frames: @@ -67,6 +68,7 @@ def upload_video(frames, user_id, category, file_name): file_name, content_type="video/mp4" # 指定MIME类型确保可在线播放[9](@ref) ) + print(file_name) return f"aida-users/{object_name}" # # 生成内存中的视频字节流 @@ -92,5 +94,5 @@ def upload_video(frames, user_id, category, file_name): if __name__ == '__main__': - images = np.random.randint(0, 256, size=(4, 512, 512, 3), dtype=np.uint8) - print(upload_video(images, user_id=89, category='test', file_name="1.mp4")) + images = np.random.randint(0, 256, size=(10, 768, 512, 3), dtype=np.uint8) + print(upload_video(images, user_id=89, category='test', file_name="1123123.mp4"))