13 lines
466 B
Python
13 lines
466 B
Python
|
from typing import Optional, List
|
||
|
|
|
||
|
|
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 = None
|
||
|
|
labels: Optional[List[int]] | None = None
|
||
|
|
box: Optional[List[int]] | None = None
|