23 lines
553 B
Python
23 lines
553 B
Python
from typing import List, Optional, Dict, Any
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class ObjectResult(BaseModel):
|
|
bbox_xyxy: List[float]
|
|
det_confidence: float
|
|
det_class_id: int
|
|
det_class_name: str
|
|
cls_class_id: int
|
|
cls_class_name: str
|
|
cls_confidence: float
|
|
|
|
|
|
class PredictResponse(BaseModel):
|
|
success: bool
|
|
image_width: int
|
|
image_height: int
|
|
num_objects: int
|
|
results: List[ObjectResult]
|
|
annotated_image_base64: Optional[str] = None
|
|
saved_outputs: Optional[Dict[str, Any]] = None
|
|
message: str = "ok" |