From 3ad724fe9f59d568d10e504e15c7a8b88a3bf39f Mon Sep 17 00:00:00 2001
From: zchengrong <124802516+zchengrong@users.noreply.github.com>
Date: Mon, 7 Apr 2025 14:24:41 +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

---
 .../utils/pose_transform_upload.py              | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/app/service/generate_image/utils/pose_transform_upload.py b/app/service/generate_image/utils/pose_transform_upload.py
index 86c3e6e..7e97a26 100644
--- a/app/service/generate_image/utils/pose_transform_upload.py
+++ b/app/service/generate_image/utils/pose_transform_upload.py
@@ -1,6 +1,7 @@
 import io
 import logging
 
+import cv2
 import imageio
 import numpy as np
 # import boto3
@@ -43,20 +44,22 @@ def upload_gif(gif_buffer, user_id, category, file_name):
 
 def upload_video(frames, user_id, category, file_name):
     try:
+        # 生成内存中的视频字节流
         video_buffer = io.BytesIO()
-        with imageio.get_writer(video_buffer, format='mp4', fps=24) as writer:
-            for frame in frames:
-                writer.append_data(frame)
-        video_buffer.seek(0)
+        with imageio.get_writer(video_buffer, format="mp4", fps=24) as writer:
+            for img in images:
+                writer.append_data(img)
+        writer.close()
+        video_bytes = video_buffer.getvalue()
 
         object_name = f'{user_id}/{category}/{file_name}'
         # 上传视频流到MinIO
         minio_client.put_object(
             bucket_name="aida-users",
             object_name=object_name,
-            data=video_buffer,
-            length=video_buffer.getbuffer().nbytes,
-            content_type='video/mp4'
+            data=io.BytesIO(video_bytes),
+            length=len(video_bytes),
+            content_type="video/mp4"
         )
         return f"aida-users/{object_name}"
     except Exception as e: