dtlpy 1.114.17__py3-none-any.whl → 1.116.6__py3-none-any.whl

dtlpy/entities/annotation_definitions/segmentation.py

@@ -1,240 +1,240 @@
-import numpy as np
-import base64
-import io
-from PIL import Image
-
-from . import BaseAnnotationDefinition
-
-from .box import Box
-from .polygon import Polygon
-
-
-class Segmentation(BaseAnnotationDefinition):
-    """
-    Segmentation annotation object
-    """
-    type = "binary"
-
-    def __init__(self, geo: np.ndarray, label: str, attributes=None, description=None, color=None):
-        super().__init__(description=description, attributes=attributes)
-        self._geo = geo
-        self._coordinates = None
-        self.label = label
-        self._color = color
-
-    @property
-    def geo(self) -> np.ndarray:
-        if self._geo is None:
-            self._geo = self.from_coordinates(self._coordinates)
-            if self._color is None:
-                color = None
-                fill_coordinates = self._geo.nonzero()
-                if len(fill_coordinates) > 0 and len(fill_coordinates[0]) > 0 and len(fill_coordinates[1]) > 0:
-                    color = self._geo[fill_coordinates[0][0]][fill_coordinates[1][0]]
-                self._color = color
-            self._geo = (self._geo[:, :, 3] > 127).astype(float)
-        return self._geo
-
-    @geo.setter
-    def geo(self, geo: np.ndarray):
-        self._geo = geo
-        self._coordinates = None
-
-    @property
-    def x(self):
-        return
-
-    @property
-    def y(self):
-        return
-
-    @property
-    def pts(self):
-        return np.where(self.geo > 0)
-
-    @property
-    def left(self):
-        left = 0
-        if len(self.pts[1]) > 0:
-            left = np.min(self.pts[1])
-        return left
-
-    @property
-    def top(self):
-        top = 0
-        if len(self.pts[0]) > 0:
-            top = np.min(self.pts[0])
-        return top
-
-    @property
-    def right(self):
-        right = 0
-        if len(self.pts[1]) > 0:
-            right = np.max(self.pts[1])
-        return right
-
-    @property
-    def bottom(self):
-        bottom = 0
-        if len(self.pts[0]) > 0:
-            bottom = np.max(self.pts[0])
-        return bottom
-
-    def show(self, image, thickness, with_text, height, width, annotation_format, color, alpha=1):
-        """
-        Show annotation as ndarray
-        :param image: empty or image to draw on
-        :param thickness:
-        :param with_text: not required
-        :param height: item height
-        :param width: item width
-        :param annotation_format: options: list(dl.ViewAnnotationOptions)
-        :param color: color
-        :param alpha: opacity value [0 1], default 1
-        :return: ndarray
-        """
-        try:
-            import cv2
-        except (ImportError, ModuleNotFoundError):
-            raise ImportError('opencv not found. Must install to perform this function')
-
-        if alpha != 1:
-            overlay = image.copy()
-        else:
-            overlay = image
-        if color is None:
-            if self._color:
-                color = self._color
-            else:
-                color = (255, 255, 255)
-        # draw annotation
-        overlay[np.where(self.geo)] = color
-
-        if not isinstance(color, int) and len(color) == 4 and color[3] != 255:
-            # add with opacity
-            image = cv2.addWeighted(src1=overlay,
-                                    alpha=alpha,
-                                    src2=image,
-                                    beta=1 - alpha,
-                                    gamma=0)
-        else:
-            image = overlay
-
-        if with_text:
-            image = self.add_text_to_image(image=image, annotation=self)
-        return image
-
-    def to_coordinates(self, color=None):
-        """
-        Convert segmentation to coordinates
-        `self._color` is ignored if `color` is provided
-        
-        :param color: color
-        :return: coordinates
-        """
-        need_encode = False
-        if color is not None and self._color is not None:
-            # if input color is not the same as the annotation's color - need to re-encode
-            if self._color != color:
-                need_encode = True
-
-        if need_encode or self._coordinates is None:
-            if color is None:
-                if self._color is not None:
-                    color = self._color
-                else:
-                    color = (255, 255, 255)
-            max_val = np.max(self.geo)
-            if max_val > 1:
-                self.geo = self.geo / max_val
-            png_ann = np.stack((color[0] * self.geo,
-                                color[1] * self.geo,
-                                color[2] * self.geo,
-                                255 * self.geo),
-                               axis=2).astype(np.uint8)
-            pil_img = Image.fromarray(png_ann)
-            buff = io.BytesIO()
-            pil_img.save(buff, format="PNG")
-            new_image_string = base64.b64encode(buff.getvalue()).decode("utf-8")
-            self._coordinates = "data:image/png;base64,%s" % new_image_string
-        return self._coordinates
-
-    def to_box(self):
-        """
-
-        :return: Box annotations list  to each separated  segmentation
-        """
-        polygons = Polygon.from_segmentation(mask=self.geo, label=self.label,
-                                             attributes=self.attributes, max_instances=None)
-
-        if not isinstance(polygons, list):
-            polygons = [polygons]
-
-        boxes = [Box(left=polygon.left,
-                     top=polygon.top,
-                     right=polygon.right,
-                     bottom=polygon.bottom,
-                     label=polygon.label,
-                     attributes=polygon.attributes) for polygon in polygons]
-
-        return boxes
-
-    @classmethod
-    def from_polygon(cls, geo, label, shape, attributes=None):
-        """
-
-        :param geo: list of x,y coordinates of the polygon ([[x,y],[x,y]...]
-        :param label: annotation's label
-        :param shape: image shape (h,w)
-        :param attributes:
-        :return:
-        """
-        try:
-            import cv2
-        except (ImportError, ModuleNotFoundError):
-            raise ImportError('opencv not found. Must install to perform this function')
-
-        thickness = -1
-
-        # plot polygon on a blank mask with thickness -1 to fill the polyline
-        mask = np.zeros(shape=shape, dtype=np.uint8)
-        mask = cv2.drawContours(image=mask,
-                                contours=[np.asarray(geo).astype('int')],
-                                contourIdx=-1,
-                                color=1,
-                                thickness=thickness)
-
-        return cls(
-            geo=mask,
-            label=label,
-            attributes=attributes,
-        )
-
-    @staticmethod
-    def from_coordinates(coordinates):
-        if isinstance(coordinates, dict):
-            data = coordinates["data"][22:]
-        elif isinstance(coordinates, str):
-            data = coordinates[22:]
-        else:
-            raise TypeError('unknown binary data type')
-        decode = base64.b64decode(data)
-        mask = np.array(Image.open(io.BytesIO(decode)))
-        return mask
-
-    @classmethod
-    def from_json(cls, _json):
-        if "coordinates" in _json:
-            coordinates = _json["coordinates"]
-        elif "data" in _json:
-            coordinates = _json["data"]
-        else:
-            raise ValueError('can not find "coordinates" or "data" in annotation. id: {}'.format(_json["id"]))
-        inst = cls(
-            geo=None,
-            label=_json["label"],
-            attributes=_json.get("attributes", None),
-            color=None
-        )
-        inst._coordinates = coordinates
-        return inst
+import numpy as np
+import base64
+import io
+from PIL import Image
+
+from . import BaseAnnotationDefinition
+
+from .box import Box
+from .polygon import Polygon
+
+
+class Segmentation(BaseAnnotationDefinition):
+    """
+    Segmentation annotation object
+    """
+    type = "binary"
+
+    def __init__(self, geo: np.ndarray, label: str, attributes=None, description=None, color=None):
+        super().__init__(description=description, attributes=attributes)
+        self._geo = geo
+        self._coordinates = None
+        self.label = label
+        self._color = color
+
+    @property
+    def geo(self) -> np.ndarray:
+        if self._geo is None:
+            self._geo = self.from_coordinates(self._coordinates)
+            if self._color is None:
+                color = None
+                fill_coordinates = self._geo.nonzero()
+                if len(fill_coordinates) > 0 and len(fill_coordinates[0]) > 0 and len(fill_coordinates[1]) > 0:
+                    color = self._geo[fill_coordinates[0][0]][fill_coordinates[1][0]]
+                self._color = color
+            self._geo = (self._geo[:, :, 3] > 127).astype(float)
+        return self._geo
+
+    @geo.setter
+    def geo(self, geo: np.ndarray):
+        self._geo = geo
+        self._coordinates = None
+
+    @property
+    def x(self):
+        return
+
+    @property
+    def y(self):
+        return
+
+    @property
+    def pts(self):
+        return np.where(self.geo > 0)
+
+    @property
+    def left(self):
+        left = 0
+        if len(self.pts[1]) > 0:
+            left = np.min(self.pts[1])
+        return left
+
+    @property
+    def top(self):
+        top = 0
+        if len(self.pts[0]) > 0:
+            top = np.min(self.pts[0])
+        return top
+
+    @property
+    def right(self):
+        right = 0
+        if len(self.pts[1]) > 0:
+            right = np.max(self.pts[1])
+        return right
+
+    @property
+    def bottom(self):
+        bottom = 0
+        if len(self.pts[0]) > 0:
+            bottom = np.max(self.pts[0])
+        return bottom
+
+    def show(self, image, thickness, with_text, height, width, annotation_format, color, alpha=1):
+        """
+        Show annotation as ndarray
+        :param image: empty or image to draw on
+        :param thickness:
+        :param with_text: not required
+        :param height: item height
+        :param width: item width
+        :param annotation_format: options: list(dl.ViewAnnotationOptions)
+        :param color: color
+        :param alpha: opacity value [0 1], default 1
+        :return: ndarray
+        """
+        try:
+            import cv2
+        except (ImportError, ModuleNotFoundError):
+            raise ImportError('opencv not found. Must install to perform this function')
+
+        if alpha != 1:
+            overlay = image.copy()
+        else:
+            overlay = image
+        if color is None:
+            if self._color:
+                color = self._color
+            else:
+                color = (255, 255, 255)
+        # draw annotation
+        overlay[np.where(self.geo)] = color
+
+        if not isinstance(color, int) and len(color) == 4 and color[3] != 255:
+            # add with opacity
+            image = cv2.addWeighted(src1=overlay,
+                                    alpha=alpha,
+                                    src2=image,
+                                    beta=1 - alpha,
+                                    gamma=0)
+        else:
+            image = overlay
+
+        if with_text:
+            image = self.add_text_to_image(image=image, annotation=self)
+        return image
+
+    def to_coordinates(self, color=None):
+        """
+        Convert segmentation to coordinates
+        `self._color` is ignored if `color` is provided
+        
+        :param color: color
+        :return: coordinates
+        """
+        need_encode = False
+        if color is not None and self._color is not None:
+            # if input color is not the same as the annotation's color - need to re-encode
+            if self._color != color:
+                need_encode = True
+
+        if need_encode or self._coordinates is None:
+            if color is None:
+                if self._color is not None:
+                    color = self._color
+                else:
+                    color = (255, 255, 255)
+            max_val = np.max(self.geo)
+            if max_val > 1:
+                self.geo = self.geo / max_val
+            png_ann = np.stack((color[0] * self.geo,
+                                color[1] * self.geo,
+                                color[2] * self.geo,
+                                255 * self.geo),
+                               axis=2).astype(np.uint8)
+            pil_img = Image.fromarray(png_ann)
+            buff = io.BytesIO()
+            pil_img.save(buff, format="PNG")
+            new_image_string = base64.b64encode(buff.getvalue()).decode("utf-8")
+            self._coordinates = "data:image/png;base64,%s" % new_image_string
+        return self._coordinates
+
+    def to_box(self):
+        """
+
+        :return: Box annotations list  to each separated  segmentation
+        """
+        polygons = Polygon.from_segmentation(mask=self.geo, label=self.label,
+                                             attributes=self.attributes, max_instances=None)
+
+        if not isinstance(polygons, list):
+            polygons = [polygons]
+
+        boxes = [Box(left=polygon.left,
+                     top=polygon.top,
+                     right=polygon.right,
+                     bottom=polygon.bottom,
+                     label=polygon.label,
+                     attributes=polygon.attributes) for polygon in polygons]
+
+        return boxes
+
+    @classmethod
+    def from_polygon(cls, geo, label, shape, attributes=None):
+        """
+
+        :param geo: list of x,y coordinates of the polygon ([[x,y],[x,y]...]
+        :param label: annotation's label
+        :param shape: image shape (h,w)
+        :param attributes:
+        :return:
+        """
+        try:
+            import cv2
+        except (ImportError, ModuleNotFoundError):
+            raise ImportError('opencv not found. Must install to perform this function')
+
+        thickness = -1
+
+        # plot polygon on a blank mask with thickness -1 to fill the polyline
+        mask = np.zeros(shape=shape, dtype=np.uint8)
+        mask = cv2.drawContours(image=mask,
+                                contours=[np.asarray(geo).astype('int')],
+                                contourIdx=-1,
+                                color=1,
+                                thickness=thickness)
+
+        return cls(
+            geo=mask,
+            label=label,
+            attributes=attributes,
+        )
+
+    @staticmethod
+    def from_coordinates(coordinates):
+        if isinstance(coordinates, dict):
+            data = coordinates["data"][22:]
+        elif isinstance(coordinates, str):
+            data = coordinates[22:]
+        else:
+            raise TypeError('unknown binary data type')
+        decode = base64.b64decode(data)
+        mask = np.array(Image.open(io.BytesIO(decode)))
+        return mask
+
+    @classmethod
+    def from_json(cls, _json):
+        if "coordinates" in _json:
+            coordinates = _json["coordinates"]
+        elif "data" in _json:
+            coordinates = _json["data"]
+        else:
+            raise ValueError('can not find "coordinates" or "data" in annotation. id: {}'.format(_json["id"]))
+        inst = cls(
+            geo=None,
+            label=_json["label"],
+            attributes=_json.get("attributes", None),
+            color=None
+        )
+        inst._coordinates = coordinates
+        return inst

dtlpy/entities/annotation_definitions/subtitle.py

@@ -1,34 +1,34 @@
-from . import BaseAnnotationDefinition
-
-
-class Subtitle(BaseAnnotationDefinition):
-    """
-        Subtitle annotation object
-    """
-    type = "subtitle"
-
-    def __init__(self, text, label, attributes=None, description=None):
-        super().__init__(description=description, attributes=attributes)
-        self.text = text
-        self.label = label
-
-    def to_coordinates(self, color):
-        return {"text": self.text}
-
-    @staticmethod
-    def from_coordinates(coordinates):
-        return coordinates["text"]
-
-    @classmethod
-    def from_json(cls, _json):
-        if "coordinates" in _json:
-            text = cls.from_coordinates(coordinates=_json["coordinates"])
-        elif "data" in _json:
-            text = cls.from_coordinates(coordinates=_json["data"])
-        else:
-            raise ValueError('Bad json, "coordinates" or "data" not found')
-        return cls(
-            text=text,
-            label=_json["label"],
-            attributes=_json.get("attributes", None)
-        )
+from . import BaseAnnotationDefinition
+
+
+class Subtitle(BaseAnnotationDefinition):
+    """
+        Subtitle annotation object
+    """
+    type = "subtitle"
+
+    def __init__(self, text, label, attributes=None, description=None):
+        super().__init__(description=description, attributes=attributes)
+        self.text = text
+        self.label = label
+
+    def to_coordinates(self, color):
+        return {"text": self.text}
+
+    @staticmethod
+    def from_coordinates(coordinates):
+        return coordinates["text"]
+
+    @classmethod
+    def from_json(cls, _json):
+        if "coordinates" in _json:
+            text = cls.from_coordinates(coordinates=_json["coordinates"])
+        elif "data" in _json:
+            text = cls.from_coordinates(coordinates=_json["data"])
+        else:
+            raise ValueError('Bad json, "coordinates" or "data" not found')
+        return cls(
+            text=text,
+            label=_json["label"],
+            attributes=_json.get("attributes", None)
+        )