diff --git a/app/api/api_design.py b/app/api/api_design.py index e5148e8..d6eec47 100644 --- a/app/api/api_design.py +++ b/app/api/api_design.py @@ -394,6 +394,7 @@ async def seg_anything(request_data: SAMRequestModel): 通过传入图片路径和点击的点坐标,返回分割后的掩码数据。 ### 参数说明: + - **user_id**:用户id 用于存储分割图 - **image_path**: 图片在服务器或云端的相对路径。 - **type**: 推理类型 - **box**: 框选矩形点位信息 @@ -406,17 +407,19 @@ async def seg_anything(request_data: SAMRequestModel): ```json point { + "user_id": 1, "image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png", "type":"point", "points": [[310, 403], [493, 375], [261, 266], [404, 484]], - "labels": [1, 1, 0, 1], + "labels": [1, 1, 0, 1] } box { + "user_id": 1, "image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png", "type":"box", - "box": [350, 286, 544, 520], + "box": [350, 286, 544, 520] } ``` """ diff --git a/app/schemas/design.py b/app/schemas/design.py index c89f4a7..6f0a633 100644 --- a/app/schemas/design.py +++ b/app/schemas/design.py @@ -4,6 +4,7 @@ from pydantic import BaseModel, Field class SAMRequestModel(BaseModel): + user_id: int = Field(..., description="用户id, 必填字段") image_path: str = Field(..., description="图片路径,必填字段") type: str = Field(..., description="推理类型,必填字段") points: Optional[List[List[float]]] = None