rvimage 0.0.1__py3-none-any.whl → 0.1.0__py3-none-any.whl

rvimage/collection_types.py

@@ -1,5 +1,5 @@
 from typing import Self
-from pydantic import BaseModel
+from pydantic import BaseModel, model_serializer, model_validator
 import numpy as np
 from rvimage.converters import (
     extract_polys_from_mask,
@@ -11,11 +11,6 @@ from rvimage.converters import (
 from rvimage.domain import BbF, BbI, Poly, enclosing_bb, find_ccs
 
 
-class GeoFig(BaseModel):
-    bbox: BbF | None = None
-    poly: Poly | None = None
-
-
 class Labelinfo(BaseModel):
     new_label: str
     labels: list[str]
@@ -26,10 +21,31 @@ class Labelinfo(BaseModel):
 
 
 class BboxAnnos(BaseModel):
-    elts: list[GeoFig]
+    elts: list[BbF | Poly]
     cat_idxs: list[int]
     selected_mask: list[bool]
 
+    @model_validator(mode="before")
+    @classmethod
+    def resolve_bb_poly(cls, data: dict) -> dict:
+        # remove the type-info from the dict
+        data["elts"] = [next(v for v in d.values()) for d in data["elts"]]
+        return data
+
+    @model_serializer()
+    def serialize_model(self):
+        elts = [
+            {"BB": elt.model_dump()}
+            if isinstance(elt, BbF)
+            else {"Poly": elt.model_dump()}
+            for elt in self.elts
+        ]
+        return {
+            "cat_idxs": self.cat_idxs,
+            "selected_mask": self.selected_mask,
+            "elts": elts,
+        }
+
     @classmethod
     def from_mask(cls, mask: np.ndarray, cat_idx: int) -> "BboxAnnos":
         """
@@ -40,7 +56,7 @@ class BboxAnnos(BaseModel):
 
         return cls(
             elts=[
-                GeoFig(poly=Poly(points=points, enclosing_bb=enclosing_bb(points)))
+                Poly(points=points, enclosing_bb=enclosing_bb(points))
                 for points in polys
             ],
             cat_idxs=cat_idxs,
@@ -60,9 +76,9 @@ class BboxAnnos(BaseModel):
     def fill_mask(self, im_mask: np.ndarray, cat_idx: int):
         fill_polys_on_mask(
             polygons=(
-                elt.poly.points
+                elt.points
                 for elt, cat_idx_ in zip(self.elts, self.cat_idxs)
-                if cat_idx == cat_idx_ and elt.poly is not None
+                if cat_idx == cat_idx_ and isinstance(elt, Poly)
             ),
             value=1,
             im_mask=im_mask,
@@ -70,9 +86,9 @@ class BboxAnnos(BaseModel):
         )
         fill_bbs_on_mask(
             bbs=(
-                elt.bbox
+                elt
                 for elt, cat_idx_ in zip(self.elts, self.cat_idxs)
-                if cat_idx == cat_idx_ and elt.bbox is not None
+                if cat_idx == cat_idx_ and isinstance(elt, BbF)
             ),
             value=1,
             im_mask=im_mask,
@@ -125,10 +141,10 @@ class BrushData(BaseModel):
 
 
 class InputAnnotationData(BaseModel):
-    bbox: BboxData
-    brush: BrushData
+    bbox: BboxData | None
+    brush: BrushData | None
 
 
 class OutputAnnotationData(BaseModel):
-    bbox: BboxAnnos
-    brush: BrushAnnos
+    bbox: BboxAnnos | None
+    brush: BrushAnnos | None

rvimage-0.1.0.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: rvimage
+Version: 0.1.0
+Summary: Pydantic types to facilitate writing prediction APIs integrated into the image annotation tool RV Image
+Project-URL: Repository, https://github.com/bertiqwerty/rvimage
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: numpy>=2.3.0
+Requires-Dist: opencv-python-headless>=4.11.0.86
+Requires-Dist: pydantic>=2.11.7
+Requires-Dist: scipy>=1.15.3
+
+# RV Image
+
+This package contains Pydantic types to facilitate writing prediction APIs integrated into the image annotation tool 
+[RV Image](https://github.com/bertiqwerty/rvimage).

rvimage-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+rvimage/collection_types.py,sha256=NvAjVrGl9HZKYSYGm71Y74opjW1llKHcS90T5x3H04Y,4071
+rvimage/converters.py,sha256=gBHgknXMuAVQnB6birZKXdFMFnbouv88QKba3JAiSPA,2578
+rvimage/domain.py,sha256=iFV5Yd3lqmSNtakwIkHZ-1LE47FiQ2PLFaFom5eRsUU,2996
+rvimage-0.1.0.dist-info/METADATA,sha256=YnR6nV_vUejxRuRBeNHfacbb3UE8lZtUCGiineBXs-4,613
+rvimage-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rvimage-0.1.0.dist-info/top_level.txt,sha256=7DS7CkxhgwOOxvcYmMMuOJ8e51SP25t0luO_dxsn2YE,8
+rvimage-0.1.0.dist-info/RECORD,,

rvimage-0.0.1.dist-info/METADATA

@@ -1,9 +0,0 @@
-Metadata-Version: 2.4
-Name: rvimage
-Version: 0.0.1
-Summary: Add your description here
-Requires-Python: >=3.12
-Description-Content-Type: text/markdown
-Requires-Dist: opencv-python-headless>=4.11.0.86
-Requires-Dist: pydantic>=2.11.7
-Requires-Dist: scipy>=1.15.3

rvimage-0.0.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-rvimage/collection_types.py,sha256=Jt0iefceOnOkCwEZFPGhb_aIeZQXG8mYwVtSGTBamV0,3492
-rvimage/converters.py,sha256=gBHgknXMuAVQnB6birZKXdFMFnbouv88QKba3JAiSPA,2578
-rvimage/domain.py,sha256=iFV5Yd3lqmSNtakwIkHZ-1LE47FiQ2PLFaFom5eRsUU,2996
-rvimage-0.0.1.dist-info/METADATA,sha256=CI8wLKpuSsOFc7LcrU0m6t6aA-9ttmui6pAJk3V35BM,260
-rvimage-0.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rvimage-0.0.1.dist-info/top_level.txt,sha256=7DS7CkxhgwOOxvcYmMMuOJ8e51SP25t0luO_dxsn2YE,8
-rvimage-0.0.1.dist-info/RECORD,,