feat(新功能): pose transform 部署
fix(修复bug): docs(文档变更): refactor(重构): test(增加测试):
This commit is contained in:
@@ -1,6 +1,7 @@
|
|||||||
import io
|
import io
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
import cv2
|
||||||
import imageio
|
import imageio
|
||||||
import numpy as np
|
import numpy as np
|
||||||
# import boto3
|
# 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):
|
def upload_video(frames, user_id, category, file_name):
|
||||||
try:
|
try:
|
||||||
|
# 生成内存中的视频字节流
|
||||||
video_buffer = io.BytesIO()
|
video_buffer = io.BytesIO()
|
||||||
with imageio.get_writer(video_buffer, format='mp4', fps=24) as writer:
|
with imageio.get_writer(video_buffer, format="mp4", fps=24) as writer:
|
||||||
for frame in frames:
|
for img in images:
|
||||||
writer.append_data(frame)
|
writer.append_data(img)
|
||||||
video_buffer.seek(0)
|
writer.close()
|
||||||
|
video_bytes = video_buffer.getvalue()
|
||||||
|
|
||||||
object_name = f'{user_id}/{category}/{file_name}'
|
object_name = f'{user_id}/{category}/{file_name}'
|
||||||
# 上传视频流到MinIO
|
# 上传视频流到MinIO
|
||||||
minio_client.put_object(
|
minio_client.put_object(
|
||||||
bucket_name="aida-users",
|
bucket_name="aida-users",
|
||||||
object_name=object_name,
|
object_name=object_name,
|
||||||
data=video_buffer,
|
data=io.BytesIO(video_bytes),
|
||||||
length=video_buffer.getbuffer().nbytes,
|
length=len(video_bytes),
|
||||||
content_type='video/mp4'
|
content_type="video/mp4"
|
||||||
)
|
)
|
||||||
return f"aida-users/{object_name}"
|
return f"aida-users/{object_name}"
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
Reference in New Issue
Block a user