20 lines
489 B
Python
20 lines
489 B
Python
from pydantic import BaseModel
|
|
from typing import List, Optional, Tuple, Dict
|
|
|
|
class Shape(BaseModel):
|
|
label: str
|
|
color: str
|
|
points: List[Tuple[float, float]]
|
|
group_id: Optional[int] = None
|
|
shape_type: str
|
|
flags: Dict[str, Optional[bool]] = {} # key는 문자열, value는 boolean 또는 None
|
|
|
|
class PredictResponse(BaseModel):
|
|
version: str
|
|
task_type: str
|
|
shapes: List[Shape]
|
|
split: str
|
|
imageHeight: int
|
|
imageWidth: int
|
|
imageDepth: int
|