feat 新增接口描述 docs页面 ,新增S3 图片get upload 操作,整理代码
fix
This commit is contained in:
@@ -8,6 +8,7 @@ import tritonclient.grpc as grpcclient
|
||||
from urllib3.exceptions import ResponseError
|
||||
|
||||
from app.core.config import *
|
||||
from app.schemas.pre_processing import DesignPreProcessingModel
|
||||
from app.service.design.utils.design_ensemble import get_keypoint_result
|
||||
from app.service.utils.oss_client import oss_get_image, oss_upload_image
|
||||
|
||||
@@ -355,3 +356,19 @@ class DesignPreprocessing:
|
||||
except Exception as e:
|
||||
logging.info(f"save keypoint cache milvus error : {e}")
|
||||
return dict(zip(KEYPOINT_RESULT_TABLE_FIELD_SET, result.reshape(12, 2).astype(int).tolist()))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
data = {
|
||||
"sketches": [
|
||||
{
|
||||
"image_category": "dress",
|
||||
"image_id": "107903",
|
||||
"image_url": "aida-sys-image/images/female/dress/0628000000.jpg"
|
||||
}
|
||||
]
|
||||
}
|
||||
request_data = DesignPreProcessingModel(sketches=data["sketches"])
|
||||
server = DesignPreprocessing()
|
||||
data = server.pipeline(image_list=request_data.sketches)
|
||||
print(data)
|
||||
|
||||
Reference in New Issue
Block a user