feat 新增接口描述 docs页面 ,新增S3 图片get upload 操作,整理代码

fix
This commit is contained in:
zhouchengrong
2024-06-26 11:15:24 +08:00
parent db3d86204f
commit d281d1e5c3
12 changed files with 30 additions and 56 deletions

View File

@@ -19,5 +19,6 @@ def rgb_to_rgba(rgb_size, rgb_image, mask):
rgba_image[:, :, 3] = alpha_channel
return rgba_image
if __name__ == '__main__':
image = open("")
image = open("")

View File

@@ -8,12 +8,14 @@
@detail :发起请求 获取推理结果
"""
import logging
import cv2
import mmcv
import numpy as np
import tritonclient.http as httpclient
import torch
import torch.nn.functional as F
import tritonclient.http as httpclient
from app.core.config import *
"""

View File

@@ -10,30 +10,13 @@
import io
import logging
# import boto3
import cv2
import numpy as np
from PIL import Image
from minio import Minio
from app.core.config import *
from app.service.utils.generate_uuid import generate_uuid
from app.service.utils.oss_client import oss_upload_image
minio_client = Minio(
MINIO_URL,
access_key=MINIO_ACCESS,
secret_key=MINIO_SECRET,
secure=MINIO_SECURE)
# s3 = boto3.client(
# 's3',
# aws_access_key_id=S3_ACCESS_KEY,
# aws_secret_access_key=S3_AWS_SECRET_ACCESS_KEY,
# region_name=S3_REGION_NAME
# )
def positioning(all_mask_shape, mask_shape, offset):
all_start = 0
@@ -176,4 +159,9 @@ def synthesis_single(front_image, back_image):
result_image.save(image_data, format='PNG')
image_data.seek(0)
image_bytes = image_data.read()
return f"aida-results/{minio_client.put_object('aida-results', f'result_{generate_uuid()}.png', io.BytesIO(image_bytes), len(image_bytes), content_type='image/png').object_name}"
# return f"aida-results/{minio_client.put_object('aida-results', f'result_{generate_uuid()}.png', io.BytesIO(image_bytes), len(image_bytes), content_type='image/png').object_name}"
# oss upload
bucket_name = 'aida-results'
object_name = f'result_{generate_uuid()}.png'
req = oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes)
return f"{bucket_name}/{object_name}"