Merge pull request #395 from Labelbox/al-1299

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

@@ -4,6 +4,7 @@
 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 +26,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 TypeError(
+                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

@@ -3,6 +3,7 @@
 import geojson
 import numpy as np
 import cv2
+from shapely.geometry import Point as SPoint
 
 from .geometry import Geometry
 
@@ -24,6 +25,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 TypeError(
+                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

@@ -4,6 +4,7 @@
 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 +31,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
+        if not isinstance(shapely_obj, SPolygon):
+            raise TypeError(
+                f"Expected Shapely Polygon. Got {shapely_obj.geom_type}")
+        obj_coords = shapely_obj.__geo_interface__['coordinates'][0]
+        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

@@ -3,6 +3,7 @@
 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 +31,24 @@ 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 TypeError(
+                f"Expected Shapely Polygon. Got {shapely_obj.geom_type}")
+
+        min_x, min_y, max_x, max_y = shapely_obj.bounds
+
+        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,