supervisely 6.73.366__py3-none-any.whl → 6.73.367__py3-none-any.whl

supervisely/geometry/bitmap_base.py

@@ -1,20 +1,34 @@
 # coding: utf-8
 from __future__ import annotations
+
+from typing import Dict, Optional, Tuple
+
 import numpy as np
-from typing import Tuple, Dict, Optional
 
-from supervisely.geometry.constants import DATA, ORIGIN, GEOMETRY_SHAPE, GEOMETRY_TYPE, \
-                                               LABELER_LOGIN, UPDATED_AT, CREATED_AT, ID, CLASS_ID
+from supervisely.geometry.constants import (
+    CLASS_ID,
+    CREATED_AT,
+    DATA,
+    GEOMETRY_SHAPE,
+    GEOMETRY_TYPE,
+    ID,
+    LABELER_LOGIN,
+    ORIGIN,
+    UPDATED_AT,
+)
 from supervisely.geometry.geometry import Geometry
 from supervisely.geometry.point_location import PointLocation
 from supervisely.geometry.rectangle import Rectangle
 from supervisely.imaging.image import resize_inter_nearest, restore_proportional_size
 
+if not hasattr(np, "bool"):
+    np.bool = np.bool_
 
-if not hasattr(np, 'bool'): np.bool = np.bool_
 
 # TODO: rename to resize_bitmap_and_origin
-def resize_origin_and_bitmap(origin: PointLocation, bitmap: np.ndarray, in_size: Tuple[int, int], out_size: Tuple[int, int]) -> Tuple[PointLocation, np.ndarray]:
+def resize_origin_and_bitmap(
+    origin: PointLocation, bitmap: np.ndarray, in_size: Tuple[int, int], out_size: Tuple[int, int]
+) -> Tuple[PointLocation, np.ndarray]:
     """
     Change PointLocation and resize numpy array to match a certain size.
 
@@ -46,7 +60,9 @@ def resize_origin_and_bitmap(origin: PointLocation, bitmap: np.ndarray, in_size:
     scaled_rows = max(round(bitmap.shape[0] * row_scale), 1)
     scaled_cols = max(round(bitmap.shape[1] * col_scale), 1)
 
-    scaled_origin = PointLocation(row=round(origin.row * row_scale), col=round(origin.col * col_scale))
+    scaled_origin = PointLocation(
+        row=round(origin.row * row_scale), col=round(origin.col * col_scale)
+    )
     scaled_bitmap = resize_inter_nearest(bitmap, (scaled_rows, scaled_cols))
     return scaled_origin, scaled_bitmap
 
@@ -74,6 +90,7 @@ class BitmapBase(Geometry):
 
     :Usage example: Example of creating and using see in :class:`Bitmap<supervisely.geometry.bitmap.Bitmap>`.
     """
+
     def __init__(
         self,
         data: np.ndarray,
@@ -85,7 +102,13 @@ class BitmapBase(Geometry):
         updated_at: Optional[str] = None,
         created_at: Optional[str] = None,
     ):
-        super().__init__(sly_id=sly_id, class_id=class_id, labeler_login=labeler_login, updated_at=updated_at, created_at=created_at)
+        super().__init__(
+            sly_id=sly_id,
+            class_id=class_id,
+            labeler_login=labeler_login,
+            updated_at=updated_at,
+            created_at=created_at,
+        )
         if origin is None:
             origin = PointLocation(row=0, col=0)
 
@@ -111,19 +134,17 @@ class BitmapBase(Geometry):
 
     @staticmethod
     def base64_2_data(s: str) -> np.ndarray:
-        """
-        """
+        """ """
         raise NotImplementedError()
 
     @staticmethod
     def data_2_base64(data: np.ndarray) -> str:
-        """
-        """
+        """ """
         raise NotImplementedError()
 
     def to_json(self) -> Dict:
         """
-        Convert the BitmapBase to a json dict. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert the BitmapBase to a json dict. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :return: Json format as a dict
         :rtype: :class:`dict`
@@ -154,7 +175,7 @@ class BitmapBase(Geometry):
         res = {
             self._impl_json_class_name(): {
                 ORIGIN: [self.origin.col, self.origin.row],
-                DATA: self.data_2_base64(self.data)
+                DATA: self.data_2_base64(self.data),
             },
             GEOMETRY_SHAPE: self.geometry_name(),
             GEOMETRY_TYPE: self.geometry_name(),
@@ -165,7 +186,7 @@ class BitmapBase(Geometry):
     @classmethod
     def from_json(cls, json_data: Dict) -> BitmapBase:
         """
-        Convert a json dict to BitmapBase. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert a json dict to BitmapBase. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :param data: Bitmap in json format as a dict.
         :type data: dict
@@ -191,11 +212,17 @@ class BitmapBase(Geometry):
         json_root_key = cls._impl_json_class_name()
         if json_root_key not in json_data:
             raise ValueError(
-                'Data must contain {} field to create MultichannelBitmap object.'.format(json_root_key))
+                "Data must contain {} field to create MultichannelBitmap object.".format(
+                    json_root_key
+                )
+            )
 
         if ORIGIN not in json_data[json_root_key] or DATA not in json_data[json_root_key]:
-            raise ValueError('{} field must contain {} and {} fields to create MultichannelBitmap object.'.format(
-                json_root_key, ORIGIN, DATA))
+            raise ValueError(
+                "{} field must contain {} and {} fields to create MultichannelBitmap object.".format(
+                    json_root_key, ORIGIN, DATA
+                )
+            )
 
         col, row = json_data[json_root_key][ORIGIN]
         data = cls.base64_2_data(json_data[json_root_key][DATA])
@@ -205,8 +232,15 @@ class BitmapBase(Geometry):
         created_at = json_data.get(CREATED_AT, None)
         sly_id = json_data.get(ID, None)
         class_id = json_data.get(CLASS_ID, None)
-        return cls(data=data, origin=PointLocation(row=row, col=col),
-                   sly_id=sly_id, class_id=class_id, labeler_login=labeler_login, updated_at=updated_at, created_at=created_at)
+        return cls(
+            data=data,
+            origin=PointLocation(row=row, col=col),
+            sly_id=sly_id,
+            class_id=class_id,
+            labeler_login=labeler_login,
+            updated_at=updated_at,
+            created_at=created_at,
+        )
 
     @property
     def origin(self) -> PointLocation:
@@ -267,7 +301,9 @@ class BitmapBase(Geometry):
             fliplr_figure = figure.fliplr((height, width))
         """
         flipped_mask = np.flip(self.data, axis=1)
-        flipped_origin = PointLocation(row=self.origin.row, col=(img_size[1] - flipped_mask.shape[1] - self.origin.col))
+        flipped_origin = PointLocation(
+            row=self.origin.row, col=(img_size[1] - flipped_mask.shape[1] - self.origin.col)
+        )
         return self.__class__(data=flipped_mask, origin=flipped_origin)
 
     def flipud(self, img_size: Tuple[int, int]) -> BitmapBase:
@@ -288,7 +324,9 @@ class BitmapBase(Geometry):
             flipud_figure = figure.flipud((height, width))
         """
         flipped_mask = np.flip(self.data, axis=0)
-        flipped_origin = PointLocation(row=(img_size[0] - flipped_mask.shape[0] - self.origin.row), col=self.origin.col)
+        flipped_origin = PointLocation(
+            row=(img_size[0] - flipped_mask.shape[0] - self.origin.row), col=self.origin.col
+        )
         return self.__class__(data=flipped_mask, origin=flipped_origin)
 
     def scale(self, factor: float) -> BitmapBase:
@@ -315,8 +353,7 @@ class BitmapBase(Geometry):
 
     @staticmethod
     def _resize_mask(mask, out_rows, out_cols):
-        """
-        """
+        """ """
         return resize_inter_nearest(mask.astype(np.uint8), (out_rows, out_cols)).astype(np.bool)
 
     def to_bbox(self) -> Rectangle:
@@ -332,4 +369,6 @@ class BitmapBase(Geometry):
 
             rectangle = figure.to_bbox()
         """
-        return Rectangle.from_array(self._data).translate(drow=self._origin.row, dcol=self._origin.col)
+        return Rectangle.from_array(self._data).translate(
+            drow=self._origin.row, dcol=self._origin.col
+        )

supervisely/geometry/cuboid.py

@@ -2,20 +2,33 @@
 
 # docs
 from __future__ import annotations
-from typing import List, Tuple, Dict, Optional
+
+from typing import Dict, List, Optional, Tuple
+
 import cv2
 import numpy as np
-from supervisely.geometry.image_rotator import ImageRotator
-from supervisely.geometry.point_location import PointLocation
-
 
-from supervisely.geometry.constants import FACES, POINTS, LABELER_LOGIN, UPDATED_AT, CREATED_AT, ID, CLASS_ID
+from supervisely.geometry.constants import (
+    CLASS_ID,
+    CREATED_AT,
+    FACES,
+    ID,
+    LABELER_LOGIN,
+    POINTS,
+    UPDATED_AT,
+)
 from supervisely.geometry.geometry import Geometry
-from supervisely.geometry.point_location import points_to_row_col_list, row_col_list_to_points
+from supervisely.geometry.image_rotator import ImageRotator
+from supervisely.geometry.point_location import (
+    PointLocation,
+    points_to_row_col_list,
+    row_col_list_to_points,
+)
 from supervisely.geometry.rectangle import Rectangle
 
+if not hasattr(np, "bool"):
+    np.bool = np.bool_
 
-if not hasattr(np, 'bool'): np.bool = np.bool_
 
 class CuboidFace:
     """
@@ -35,6 +48,7 @@ class CuboidFace:
 
         edge = CuboidFace(0, 1, 2, 3)
     """
+
     def __init__(self, a: int, b: int, c: int, d: int):
         self._a = a
         self._b = b
@@ -43,7 +57,7 @@ class CuboidFace:
 
     def to_json(self) -> List[int]:
         """
-        Convert the CuboidFace to list. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert the CuboidFace to list. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :return: List of integers
         :rtype: :class:`List[int, int, int, int]`
@@ -61,7 +75,7 @@ class CuboidFace:
     @classmethod
     def from_json(cls, data: List[int]) -> CuboidFace:
         """
-        Convert list of integers to CuboidFace. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert list of integers to CuboidFace. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :param data: List of integers.
         :type data: List[int, int, int, int]
@@ -75,31 +89,29 @@ class CuboidFace:
             new_edge = CuboidFace.from_json([0, 1, 2, 3])
         """
         if len(data) != 4:
-            raise ValueError(f'CuboidFace JSON data must have exactly 4 indices, instead got {len(data)!r}.')
+            raise ValueError(
+                f"CuboidFace JSON data must have exactly 4 indices, instead got {len(data)!r}."
+            )
         return cls(data[0], data[1], data[2], data[3])
 
     @property
     def a(self):
-        """
-        """
+        """ """
         return self._a
 
     @property
     def b(self):
-        """
-        """
+        """ """
         return self._b
 
     @property
     def c(self):
-        """
-        """
+        """ """
         return self._c
 
     @property
     def d(self):
-        """
-        """
+        """ """
         return self._d
 
     def tolist(self) -> List[int]:
@@ -151,28 +163,44 @@ class Cuboid(Geometry):
         pl_nodes = (sly.PointLocation(node[0], node[1]) for node in nodes)
         figure = sly.Cuboid(pl_nodes, edges)
     """
+
     @staticmethod
     def geometry_name():
-        """
-        """
-        return 'cuboid'
-
-    def __init__(self, points: List[PointLocation], faces: List[CuboidFace], sly_id: Optional[int] = None, class_id: Optional[int] = None,
-                 labeler_login: Optional[int] = None, updated_at: Optional[str] = None, created_at: Optional[str] = None):
-
-        super().__init__(sly_id=sly_id, class_id=class_id, labeler_login=labeler_login, updated_at=updated_at, created_at=created_at)
+        """ """
+        return "cuboid"
+
+    def __init__(
+        self,
+        points: List[PointLocation],
+        faces: List[CuboidFace],
+        sly_id: Optional[int] = None,
+        class_id: Optional[int] = None,
+        labeler_login: Optional[int] = None,
+        updated_at: Optional[str] = None,
+        created_at: Optional[str] = None,
+    ):
+
+        super().__init__(
+            sly_id=sly_id,
+            class_id=class_id,
+            labeler_login=labeler_login,
+            updated_at=updated_at,
+            created_at=created_at,
+        )
 
         points = list(points)
         faces = list(faces)
 
         if len(faces) != 3:
-            raise ValueError(f'A cuboid must have exactly 3 faces. Instead got {len(faces)} faces.')
+            raise ValueError(f"A cuboid must have exactly 3 faces. Instead got {len(faces)} faces.")
 
         for face in faces:
             for point_idx in (face.a, face.b, face.c, face.d):
                 if point_idx >= len(points) or point_idx < 0:
-                    raise ValueError(f'Point index is out of bounds for cuboid face. Got {len(points)!r} points, but '
-                                     f'the index is {point_idx!r}.')
+                    raise ValueError(
+                        f"Point index is out of bounds for cuboid face. Got {len(points)!r} points, but "
+                        f"the index is {point_idx!r}."
+                    )
 
         self._points = points
         self._faces = faces
@@ -199,7 +227,7 @@ class Cuboid(Geometry):
 
     def to_json(self) -> Dict:
         """
-        Convert the Cuboid to a json dict. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert the Cuboid to a json dict. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :return: Json format as a dict
         :rtype: :class:`dict`
@@ -236,7 +264,7 @@ class Cuboid(Geometry):
         """
         packed_obj = {
             POINTS: points_to_row_col_list(self._points, flip_row_col_order=True),
-            FACES: [face.to_json() for face in self._faces]
+            FACES: [face.to_json() for face in self._faces],
         }
         self._add_creation_info(packed_obj)
         return packed_obj
@@ -244,7 +272,7 @@ class Cuboid(Geometry):
     @classmethod
     def from_json(cls, data: Dict) -> Cuboid:
         """
-        Convert a json dict to Cuboid. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert a json dict to Cuboid. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :param data: Cuboid in json format as a dict.
         :type data: dict
@@ -278,7 +306,7 @@ class Cuboid(Geometry):
         """
         for k in [POINTS, FACES]:
             if k not in data:
-                raise ValueError(f'Field {k!r} not found in Cuboid JSON data.')
+                raise ValueError(f"Field {k!r} not found in Cuboid JSON data.")
 
         points = row_col_list_to_points(data[POINTS], flip_row_col_order=True)
         faces = [CuboidFace.from_json(face_json) for face_json in data[FACES]]
@@ -288,8 +316,15 @@ class Cuboid(Geometry):
         created_at = data.get(CREATED_AT, None)
         sly_id = data.get(ID, None)
         class_id = data.get(CLASS_ID, None)
-        return cls(points=points, faces=faces,
-                   sly_id=sly_id, class_id=class_id, labeler_login=labeler_login, updated_at=updated_at, created_at=created_at)
+        return cls(
+            points=points,
+            faces=faces,
+            sly_id=sly_id,
+            class_id=class_id,
+            labeler_login=labeler_login,
+            updated_at=updated_at,
+            created_at=created_at,
+        )
 
     def crop(self, rect: Rectangle) -> List[Cuboid]:
         """
@@ -314,12 +349,14 @@ class Cuboid(Geometry):
             crop_figures = figure.crop(sly.Rectangle(1, 1, 1500, 1550))
         """
         is_all_nodes_inside = all(
-            rect.contains_point_location(self._points[p]) for face in self._faces for p in face.tolist())
+            rect.contains_point_location(self._points[p])
+            for face in self._faces
+            for p in face.tolist()
+        )
         return [self] if is_all_nodes_inside else []
 
     def _transform(self, transform_fn):
-        """
-        """
+        """ """
         return Cuboid(points=[transform_fn(p) for p in self.points], faces=self.faces)
 
     def rotate(self, rotator: ImageRotator) -> Cuboid:
@@ -489,8 +526,7 @@ class Cuboid(Geometry):
         return self._transform(lambda p: p.flipud(img_size))
 
     def _draw_impl(self, bitmap: np.ndarray, color, thickness=1, config=None):
-        """
-        """
+        """ """
         bmp_to_draw = np.zeros(bitmap.shape[:2], np.uint8)
         for contour in self._contours_list():
             cv2.fillPoly(bmp_to_draw, pts=[np.array(contour, dtype=np.int32)], color=1)
@@ -498,16 +534,18 @@ class Cuboid(Geometry):
         bitmap[bool_mask] = color
 
     def _draw_contour_impl(self, bitmap, color, thickness=1, config=None):
-        """
-        """
+        """ """
         contours_np_list = [np.array(contour, dtype=np.int32) for contour in self._contours_list()]
         cv2.polylines(bitmap, pts=contours_np_list, isClosed=True, color=color, thickness=thickness)
 
     def _contours_list(self):
-        """
-        """
-        return [points_to_row_col_list([self._points[idx] for idx in face.tolist()], flip_row_col_order=True)
-                for face in self._faces]
+        """ """
+        return [
+            points_to_row_col_list(
+                [self._points[idx] for idx in face.tolist()], flip_row_col_order=True
+            )
+            for face in self._faces
+        ]
 
     def to_bbox(self) -> Rectangle:
         """
@@ -530,11 +568,20 @@ class Cuboid(Geometry):
 
             rectangle = figure.to_bbox()
         """
-        points_np = np.array([[self._points[p].row, self._points[p].col]
-                              for face in self._faces for p in face.tolist()])
+        points_np = np.array(
+            [
+                [self._points[p].row, self._points[p].col]
+                for face in self._faces
+                for p in face.tolist()
+            ]
+        )
         rows, cols = points_np[:, 0], points_np[:, 1]
-        return Rectangle(top=round(min(rows).item()), left=round(min(cols).item()), bottom=round(max(rows).item()),
-                         right=round(max(cols).item()))
+        return Rectangle(
+            top=round(min(rows).item()),
+            left=round(min(cols).item()),
+            bottom=round(max(rows).item()),
+            right=round(max(cols).item()),
+        )
 
     @property
     def area(self) -> float:

supervisely/geometry/cuboid_2d.py

@@ -151,7 +151,7 @@ class Cuboid2d(GraphNodes):
     @classmethod
     def from_json(cls, data: Dict[str, Dict]) -> Cuboid2d:
         """
-        Convert a json dict to Cuboid2d. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert a json dict to Cuboid2d. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :param data: Cuboid2d in json format as a dict.
         :type data: Dict[str, Dict]
@@ -214,7 +214,7 @@ class Cuboid2d(GraphNodes):
 
     def to_json(self) -> Dict[str, Dict]:
         """
-        Convert the Cuboid2d to list. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert the Cuboid2d to list. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :return: Json format as a dict
         :rtype: Dict[str, Dict]

supervisely/geometry/graph.py

@@ -96,7 +96,7 @@ class Node(JsonSerializable):
     @classmethod
     def from_json(cls, data: Dict) -> Node:
         """
-        Convert a json dict to Node. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert a json dict to Node. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :param data: Node in json format as a dict.
         :type data: dict
@@ -120,7 +120,7 @@ class Node(JsonSerializable):
 
     def to_json(self) -> Dict:
         """
-        Convert the Node to a json dict. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert the Node to a json dict. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :return: Json format as a dict
         :rtype: :class:`dict`
@@ -240,7 +240,7 @@ class GraphNodes(Geometry):
     @classmethod
     def from_json(cls, data: Dict[str, Dict]) -> GraphNodes:
         """
-        Convert a json dict to GraphNodes. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert a json dict to GraphNodes. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :param data: GraphNodes in json format as a dict.
         :type data: dict
@@ -286,7 +286,7 @@ class GraphNodes(Geometry):
 
     def to_json(self) -> Dict[str, Dict]:
         """
-        Convert the GraphNodes to list. Read more about `Supervisely format <https://docs.supervise.ly/data-organization/00_ann_format_navi>`_.
+        Convert the GraphNodes to list. Read more about `Supervisely format <https://docs.supervisely.com/data-organization/00_ann_format_navi>`_.
 
         :return: Json format as a dict
         :rtype: :class:`dict`

supervisely/geometry/mask_3d.py

@@ -461,7 +461,7 @@ class Mask3D(Geometry):
             import os
             import nrrd
 
-            address = 'https://app.supervise.ly/'
+            address = 'https://app.supervisely.com/'
             token = 'Your Supervisely API Token'
             api = sly.Api(address, token)