feat: seg any thing 新增box模式
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
This commit is contained in:
@@ -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]
|
||||
}
|
||||
```
|
||||
"""
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user