object-segmentation/src/database/models.py

"""
Data models for the microscopy object detection application.
These dataclasses represent the database entities.
"""

from dataclasses import dataclass
from datetime import datetime
from typing import Optional, Dict, Tuple, List


@dataclass
class Model:
    """Represents a trained model."""

    id: Optional[int]
    model_name: str
    model_version: str
    model_path: str
    base_model: str
    created_at: datetime
    training_params: Optional[Dict]
    metrics: Optional[Dict]


@dataclass
class Image:
    """Represents an image in the database."""

    id: Optional[int]
    relative_path: str
    filename: str
    width: int
    height: int
    captured_at: Optional[datetime]
    added_at: datetime
    checksum: Optional[str]


@dataclass
class Detection:
    """Represents a detection result."""

    id: Optional[int]
    image_id: int
    model_id: int
    class_name: str
    bbox: Tuple[float, float, float, float]  # (x_min, y_min, x_max, y_max)
    confidence: float
    segmentation_mask: Optional[
        List[List[float]]
    ]  # List of polygon coordinates [[x1,y1], [x2,y2], ...]
    detected_at: datetime
    metadata: Optional[Dict]


@dataclass
class Annotation:
    """Represents a manual annotation."""

    id: Optional[int]
    image_id: int
    class_name: str
    bbox: Tuple[float, float, float, float]  # (x_min, y_min, x_max, y_max)
    segmentation_mask: Optional[
        List[List[float]]
    ]  # List of polygon coordinates [[x1,y1], [x2,y2], ...]
    annotator: str
    created_at: datetime
    verified: bool