Making it installabel package and switching to segmentation mode

src/database/models.py

@@ -5,7 +5,7 @@ These dataclasses represent the database entities.
 
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Optional, Dict, Tuple
+from typing import Optional, Dict, Tuple, List
 
 
 @dataclass
@@ -46,6 +46,9 @@ class Detection:
     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]
 
@@ -58,6 +61,9 @@ class Annotation:
     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