feat(新功能): 模特编辑功能参数修改
fix(修复bug): docs(文档变更): refactor(重构): test(增加测试):
This commit is contained in:
@@ -17,14 +17,17 @@ def mannequins_edit(request_data: MannequinModel):
|
|||||||
模特腿长调整
|
模特腿长调整
|
||||||
创建一个具有以下参数的请求体:
|
创建一个具有以下参数的请求体:
|
||||||
- **mannequins**: mannequins url等信息
|
- **mannequins**: mannequins url等信息
|
||||||
- **scale**: 大腿小腿比例
|
- **resize_pixel**: 拉伸像素量
|
||||||
- **bucket_name**: bucket name
|
- **bucket_name**: bucket name
|
||||||
- **mannequin_name**: 模特名称
|
- **mannequin_name**: 模特名称
|
||||||
|
- **top**: 拉伸y轴点位
|
||||||
|
- **bottom**: 拉伸y轴点位
|
||||||
|
|
||||||
|
|
||||||
示例参数:
|
示例参数:
|
||||||
- **{
|
- **{
|
||||||
"mannequins": "aida-sys-image/models/male/dc36ce58-46c3-4b6f-8787-5ca7d6fc26e6.png",
|
"mannequins": "aida-sys-image/models/male/dc36ce58-46c3-4b6f-8787-5ca7d6fc26e6.png",
|
||||||
"scale": 0.75,
|
"resize_pixel": -50,
|
||||||
"bucket_name": "test",
|
"bucket_name": "test",
|
||||||
"mannequin_name": "mannequin_name",
|
"mannequin_name": "mannequin_name",
|
||||||
"top" : 270,
|
"top" : 270,
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ from pydantic import BaseModel
|
|||||||
|
|
||||||
class MannequinModel(BaseModel):
|
class MannequinModel(BaseModel):
|
||||||
mannequins: str
|
mannequins: str
|
||||||
scale: float
|
resize_pixel: float
|
||||||
bucket_name: str
|
bucket_name: str
|
||||||
mannequin_name: str
|
mannequin_name: str
|
||||||
top: int
|
top: int
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET
|
|||||||
|
|
||||||
class MannequinEditService():
|
class MannequinEditService():
|
||||||
def __init__(self, request_data):
|
def __init__(self, request_data):
|
||||||
self.scale = request_data.scale
|
self.resize_pixel = request_data.resize_pixel
|
||||||
self.top = request_data.top
|
self.top = request_data.top
|
||||||
self.bottom = request_data.bottom
|
self.bottom = request_data.bottom
|
||||||
self.image = oss_get_image(oss_client=minio_client, bucket=request_data.mannequins.split('/')[0], object_name=request_data.mannequins[request_data.mannequins.find('/') + 1:], data_type="cv2")
|
self.image = oss_get_image(oss_client=minio_client, bucket=request_data.mannequins.split('/')[0], object_name=request_data.mannequins[request_data.mannequins.find('/') + 1:], data_type="cv2")
|
||||||
@@ -47,7 +47,7 @@ class MannequinEditService():
|
|||||||
part_bottom = self.bgr[bottom:, :]
|
part_bottom = self.bgr[bottom:, :]
|
||||||
part_bottom_alpha = self.alpha[bottom:, :]
|
part_bottom_alpha = self.alpha[bottom:, :]
|
||||||
|
|
||||||
new_height = int((bottom - top) * self.scale)
|
new_height = int((bottom - top) + self.resize_pixel)
|
||||||
|
|
||||||
resized_thigh = cv2.resize(part_resize, (self.w, new_height), interpolation=cv2.INTER_LINEAR)
|
resized_thigh = cv2.resize(part_resize, (self.w, new_height), interpolation=cv2.INTER_LINEAR)
|
||||||
resized_thigh_alpha = cv2.resize(part_resize_alpha, (self.w, new_height), interpolation=cv2.INTER_LINEAR)
|
resized_thigh_alpha = cv2.resize(part_resize_alpha, (self.w, new_height), interpolation=cv2.INTER_LINEAR)
|
||||||
@@ -68,7 +68,7 @@ class MannequinEditService():
|
|||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
request_data = MannequinModel(
|
request_data = MannequinModel(
|
||||||
mannequins="aida-sys-image/models/male/dc36ce58-46c3-4b6f-8787-5ca7d6fc26e6.png",
|
mannequins="aida-sys-image/models/male/dc36ce58-46c3-4b6f-8787-5ca7d6fc26e6.png",
|
||||||
scale=0.1,
|
resize_pixel=-100,
|
||||||
bucket_name="test",
|
bucket_name="test",
|
||||||
mannequin_name="mannequin_name",
|
mannequin_name="mannequin_name",
|
||||||
top=270,
|
top=270,
|
||||||
|
|||||||
Reference in New Issue
Block a user