addition of from shapely function for annotation types

Author: jtsodapop

labelbox/data/annotation_types/data/tiled_image.py

@@ -15,11 +15,7 @@
 from pydantic import BaseModel, validator
 from pydantic.class_validators import root_validator
 
-from labelbox.data.annotation_types.geometry.polygon import Polygon
-from labelbox.data.annotation_types.geometry.point import Point
-from labelbox.data.annotation_types.geometry.line import Line
-from labelbox.data.annotation_types.geometry.rectangle import Rectangle
-from ..geometry import Point
+from labelbox.data.annotation_types import Rectangle, Point, Line, Polygon
 from .base_data import BaseData
 from .raster import RasterData
 

labelbox/data/annotation_types/geometry/line.py

@@ -1,9 +1,11 @@
+from __future__ import annotations
 from typing import List, Optional, Union, Tuple
 
 import geojson
 import numpy as np
 import cv2
 from pydantic import validator
+from shapely.geometry import LineString as SLineString
 
 from .point import Point
 from .geometry import Geometry
@@ -25,6 +27,17 @@ def geometry(self) -> geojson.MultiLineString:
         return geojson.MultiLineString(
             [[[point.x, point.y] for point in self.points]])
 
+    @classmethod
+    def from_shapely(cls, shapely_obj: SLineString) -> Line:
+        """Transforms a shapely object."""
+        if not isinstance(shapely_obj, SLineString):
+            raise ValueError(
+                f"Expected Shapely Line. Got {shapely_obj.geom_type}")
+
+        obj_coords = shapely_obj.__geo_interface__['coordinates']
+        return Line(
+            points=[Point(x=coords[0], y=coords[1]) for coords in obj_coords])
+
     def draw(self,
              height: Optional[int] = None,
              width: Optional[int] = None,

labelbox/data/annotation_types/geometry/point.py

@@ -1,8 +1,10 @@
+from __future__ import annotations
 from typing import Optional, Tuple, Union
 
 import geojson
 import numpy as np
 import cv2
+from shapely.geometry import Point as SPoint
 
 from .geometry import Geometry
 
@@ -24,6 +26,16 @@ class Point(Geometry):
     def geometry(self) -> geojson.Point:
         return geojson.Point((self.x, self.y))
 
+    @classmethod
+    def from_shapely(cls, shapely_obj: SPoint) -> Point:
+        """Transforms a shapely object."""
+        if not isinstance(shapely_obj, SPoint):
+            raise ValueError(
+                f"Expected Shapely Point. Got {shapely_obj.geom_type}")
+
+        obj_coords = shapely_obj.__geo_interface__['coordinates']
+        return Point(x=obj_coords[0], y=obj_coords[1])
+
     def draw(self,
              height: Optional[int] = None,
              width: Optional[int] = None,

labelbox/data/annotation_types/geometry/polygon.py

@@ -1,9 +1,11 @@
+from __future__ import annotations
 from typing import List, Optional, Union, Tuple
 
 import cv2
 import geojson
 import numpy as np
 from pydantic import validator
+from shapely.geometry import Polygon as SPolygon
 
 from .geometry import Geometry
 from .point import Point
@@ -30,6 +32,17 @@ def geometry(self) -> geojson.Polygon:
             self.points.append(self.points[0])
         return geojson.Polygon([[(point.x, point.y) for point in self.points]])
 
+    @classmethod
+    def from_shapely(cls, shapely_obj: SPolygon) -> Polygon:
+        """Transforms a shapely object."""
+        #we only consider 0th index because we only allow for filled polygons
+        obj_coords = shapely_obj.__geo_interface__['coordinates'][0]
+        if not isinstance(shapely_obj, SPolygon):
+            raise ValueError(
+                f"Expected Shapely Polygon. Got {shapely_obj.geom_type}")
+        return Polygon(
+            points=[Point(x=coords[0], y=coords[1]) for coords in obj_coords])
+
     def draw(self,
              height: Optional[int] = None,
              width: Optional[int] = None,

labelbox/data/annotation_types/geometry/rectangle.py

@@ -1,8 +1,10 @@
+from __future__ import annotations
 from typing import Optional, Union, Tuple
 
 import cv2
 import geojson
 import numpy as np
+from shapely.geometry import Polygon as SPolygon
 
 from .geometry import Geometry
 from .point import Point
@@ -30,6 +32,28 @@ def geometry(self) -> geojson.geometry.Geometry:
             [self.start.x, self.start.y],
         ]])
 
+    @classmethod
+    def from_shapely(cls, shapely_obj: SPolygon) -> Rectangle:
+        """Transforms a shapely object.
+        
+        If the provided shape is a non-rectangular polygon, a rectangle will be
+        returned based on the min and max x,y values."""
+        if not isinstance(shapely_obj, SPolygon):
+            raise ValueError(
+                f"Expected Shapely Polygon. Got {shapely_obj.geom_type}")
+
+        #we only consider 0th index because we only allow for filled polygons
+        obj_coords = np.array(shapely_obj.__geo_interface__['coordinates'][0])
+
+        min_x, max_x = np.min(obj_coords[:, 0]), np.max(obj_coords[:, 0])
+        min_y, max_y = np.min(obj_coords[:, 1]), np.max(obj_coords[:, 1])
+
+        start = [min_x, min_y]
+        end = [max_x, max_y]
+
+        return Rectangle(start=Point(x=start[0], y=start[1]),
+                         end=Point(x=end[0], y=end[1]))
+
     def draw(self,
              height: Optional[int] = None,
              width: Optional[int] = None,