Compare commits
4 Commits
0ac5a4e0a8
...
dev-ltx
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
fb46a9521d | ||
|
|
b90688f835 | ||
| 7e30779aec | |||
| f7294f5966 |
@@ -394,7 +394,10 @@ async def seg_anything(request_data: SAMRequestModel):
|
|||||||
通过传入图片路径和点击的点坐标,返回分割后的掩码数据。
|
通过传入图片路径和点击的点坐标,返回分割后的掩码数据。
|
||||||
|
|
||||||
### 参数说明:
|
### 参数说明:
|
||||||
|
- **user_id**:用户id 用于存储分割图
|
||||||
- **image_path**: 图片在服务器或云端的相对路径。
|
- **image_path**: 图片在服务器或云端的相对路径。
|
||||||
|
- **type**: 推理类型
|
||||||
|
- **box**: 框选矩形点位信息
|
||||||
- **points**: 交互点的坐标列表。每个点为 [x, y] 像素格式。
|
- **points**: 交互点的坐标列表。每个点为 [x, y] 像素格式。
|
||||||
- **labels**: 坐标点的属性标签,必须与 points 长度一致:
|
- **labels**: 坐标点的属性标签,必须与 points 长度一致:
|
||||||
- 1: **前景点** (代表想要分割出的区域)
|
- 1: **前景点** (代表想要分割出的区域)
|
||||||
@@ -402,11 +405,22 @@ async def seg_anything(request_data: SAMRequestModel):
|
|||||||
|
|
||||||
### 请求体示例:
|
### 请求体示例:
|
||||||
```json
|
```json
|
||||||
|
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",
|
||||||
"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
|
||||||
|
{
|
||||||
|
"user_id": 1,
|
||||||
|
"image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png",
|
||||||
|
"type":"box",
|
||||||
|
"box": [350, 286, 544, 520]
|
||||||
|
}
|
||||||
```
|
```
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -141,6 +141,9 @@ router = APIRouter()
|
|||||||
async def startup_event():
|
async def startup_event():
|
||||||
"""启动时初始化增量监听任务"""
|
"""启动时初始化增量监听任务"""
|
||||||
try:
|
try:
|
||||||
|
# 屏蔽 apscheduler 的 INFO 日志
|
||||||
|
logging.getLogger("apscheduler").setLevel(logging.WARNING)
|
||||||
|
|
||||||
# 确保 Milvus 集合已创建(若已存在则直接返回)
|
# 确保 Milvus 集合已创建(若已存在则直接返回)
|
||||||
try:
|
try:
|
||||||
create_collection()
|
create_collection()
|
||||||
|
|||||||
@@ -1,10 +1,15 @@
|
|||||||
from pydantic import BaseModel
|
from typing import List, Optional
|
||||||
|
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
class SAMRequestModel(BaseModel):
|
class SAMRequestModel(BaseModel):
|
||||||
image_path: str
|
user_id: int = Field(..., description="用户id, 必填字段")
|
||||||
points: list[list[float]]
|
image_path: str = Field(..., description="图片路径,必填字段")
|
||||||
labels: list[int]
|
type: str = Field(..., description="推理类型,必填字段")
|
||||||
|
points: Optional[List[List[float]]] = None
|
||||||
|
labels: Optional[List[int]] = None
|
||||||
|
box: Optional[List[int]] = None
|
||||||
|
|
||||||
|
|
||||||
class DesignModel(BaseModel):
|
class DesignModel(BaseModel):
|
||||||
|
|||||||
@@ -308,6 +308,10 @@ class IncrementalListener:
|
|||||||
|
|
||||||
def start_background_listener(scheduler: BackgroundScheduler):
|
def start_background_listener(scheduler: BackgroundScheduler):
|
||||||
"""将增量监听任务注册到后台调度器"""
|
"""将增量监听任务注册到后台调度器"""
|
||||||
|
# 降低 apscheduler 的日志级别,避免大量刷屏
|
||||||
|
logging.getLogger('apscheduler.executors.default').setLevel(logging.WARNING)
|
||||||
|
logging.getLogger('apscheduler.scheduler').setLevel(logging.WARNING)
|
||||||
|
|
||||||
listener = IncrementalListener()
|
listener = IncrementalListener()
|
||||||
scheduler.add_job(
|
scheduler.add_job(
|
||||||
listener.process_once,
|
listener.process_once,
|
||||||
|
|||||||
Reference in New Issue
Block a user