aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: seg any thing 新增box模式
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
Details

Browse Source
This commit is contained in:
zcr
2026-01-13 12:43:30 +08:00
parent f7294f5966
commit 7e30779aec
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/api/api_design.py
View File

@@ -394,6 +394,7 @@ async def seg_anything(request_data: SAMRequestModel):
通过传入图片路径和点击的点坐标，返回分割后的掩码数据。 通过传入图片路径和点击的点坐标，返回分割后的掩码数据。
### 参数说明: ### 参数说明:
- **user_id**:用户id 用于存储分割图
- **image_path**: 图片在服务器或云端的相对路径。 - **image_path**: 图片在服务器或云端的相对路径。
- **type**: 推理类型 - **type**: 推理类型
- **box**: 框选矩形点位信息 - **box**: 框选矩形点位信息
@@ -406,17 +407,19 @@ async def seg_anything(request_data: SAMRequestModel):
```json ```json
point point
{ {
"user_id": 1,
"image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png", "image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png",
"type":"point", "type":"point",
"points": [[310, 403], [493, 375], [261, 266], [404, 484]], "points": [[310, 403], [493, 375], [261, 266], [404, 484]],
"labels": [1, 1, 0, 1], "labels": [1, 1, 0, 1]
} }
box box
{ {
"user_id": 1,
"image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png", "image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png",
"type":"box", "type":"box",
"box": [350, 286, 544, 520], "box": [350, 286, 544, 520]
} }
``` ```
""" """

1
app/schemas/design.py
View File

@@ -4,6 +4,7 @@ from pydantic import BaseModel, Field
class SAMRequestModel(BaseModel): class SAMRequestModel(BaseModel):
user_id: int = Field(..., description="用户id, 必填字段")
image_path: str = Field(..., description="图片路径，必填字段") image_path: str = Field(..., description="图片路径，必填字段")
type: str = Field(..., description="推理类型，必填字段") type: str = Field(..., description="推理类型，必填字段")
points: Optional[List[List[float]]] = None points: Optional[List[List[float]]] = None