supervisely 6.73.350__py3-none-any.whl → 6.73.352__py3-none-any.whl

supervisely/nn/inference/tracking/point_tracking.py

@@ -369,6 +369,8 @@ class PointTracking(BaseTracking):
                 if video_interface.global_stop_indicatior:
                     return
 
+                geometries = [video_interface._crop_geometry(g) for g in geometries]
+                geometries = [g for g in geometries if g is not None]
                 geometries = [{"type": g.geometry_name(), "data": g.to_json()} for g in geometries]
                 predictions.append(geometries)
 

supervisely/nn/inference/tracking/tracker_interface.py

@@ -18,6 +18,7 @@ from supervisely.geometry.helpers import deserialize_geometry
 from supervisely.geometry.point import Point
 from supervisely.geometry.polygon import Polygon
 from supervisely.geometry.polyline import Polyline
+from supervisely.geometry.rectangle import Rectangle
 from supervisely.nn.inference.cache import InferenceImageCache
 from supervisely.sly_logger import logger
 from supervisely.video_annotation.key_id_map import KeyIdMap
@@ -42,6 +43,9 @@ class TrackerInterface:
 
         self.track_id = context["trackId"]
         self.video_id = context["videoId"]
+        self.frame_width = context.get("frameWidth", None)
+        self.frame_height = context.get("frameHeight", None)
+        self._video_info = None
         self.object_ids = list(context["objectIds"])
         self.figure_ids = list(context["figureIds"])
         self.direction = context["direction"]
@@ -75,6 +79,12 @@ class TrackerInterface:
             self._load_frames_to_hot_cache()
             self._load_frames()
 
+    @property
+    def video_info(self):
+        if self._video_info is None:
+            self._video_info = self.api.video.get_info_by_id(self.video_id)
+        return self._video_info
+
     def add_object_geometries(self, geometries: List[Geometry], object_id: int, start_fig: int):
         for frame, geometry in zip(self._cur_frames_indexes[1:], geometries):
             if self.global_stop_indicatior:
@@ -115,6 +125,17 @@ class TrackerInterface:
                 self.clear_cache()
                 return
 
+    def _crop_geometry(self, geometry: Geometry):
+        h, w = self.frame_height, self.frame_width
+        if h is None or w is None:
+            h = self.video_info.frame_height
+            w = self.video_info.frame_width
+        rect = Rectangle.from_size((h, w))
+        cropped = geometry.crop(rect)
+        if len(cropped) == 0:
+            return None
+        return cropped[0]
+
     def add_object_geometries_on_frames(
         self,
         geometries: List[Geometry],
@@ -131,6 +152,13 @@ class TrackerInterface:
         geometries_by_object = _split(geometries, object_ids, frame_indexes)
 
         for object_id, geometries_frame_indexes in geometries_by_object.items():
+            for i, (geometry, frame_index) in enumerate(geometries_frame_indexes):
+                geometries_frame_indexes[i] = (self._crop_geometry(geometry), frame_index)
+            geometries_frame_indexes = [
+                (geometry, frame_index)
+                for geometry, frame_index in geometries_frame_indexes
+                if geometry is not None
+            ]
             figures_json = [
                 {
                     ApiField.OBJECT_ID: object_id,

{supervisely-6.73.350.dist-info → supervisely-6.73.352.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.350
+Version: 6.73.352
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely
@@ -40,7 +40,7 @@ Requires-Dist: stringcase<2.0.0,>=1.2.0
 Requires-Dist: python-magic<1.0.0,>=0.4.25
 Requires-Dist: trimesh<=4.5.0,>=3.11.2
 Requires-Dist: uvicorn[standard]<1.0.0,>=0.18.2
-Requires-Dist: pydantic<=2.8.2,>=1.7.4
+Requires-Dist: pydantic<=2.11.3,>=1.7.4
 Requires-Dist: anyio<=4.2.0,>=3.7.1
 Requires-Dist: fastapi<=0.109.0,>=0.79.0
 Requires-Dist: websockets<=13.1,>=10.3

{supervisely-6.73.350.dist-info → supervisely-6.73.352.dist-info}/RECORD

@@ -922,9 +922,9 @@ supervisely/nn/inference/tracking/bbox_tracking.py,sha256=sPE-jZjSKf5QuMXmYUOkP1
 supervisely/nn/inference/tracking/functional.py,sha256=LpVu2gvOOpr9D_uvwTPZey1wUCAhV-E20RPKmCSIrK4,1774
 supervisely/nn/inference/tracking/mask_tracking.py,sha256=1G0yHMw5BSzDGD4Ps-yRhr50ZeB88pMY6b-rmg2FFHw,25576
 supervisely/nn/inference/tracking/object_tracking_3d.py,sha256=Kqvx1qe1G8F1VtdBiy2HJ251rJU6s3LWhj0ZedhrmUw,4327
-supervisely/nn/inference/tracking/point_tracking.py,sha256=I0kT_YGqOudyX4LMSZfnWkB1UnUQ9ucawTyR-tFonc8,24644
+supervisely/nn/inference/tracking/point_tracking.py,sha256=jQduMpFrntITDRCNb2q93TwSfkge4T_2nMJ71iJtzqM,24799
 supervisely/nn/inference/tracking/tracker3d_interface.py,sha256=7yIkNO9rgkzQuyXUUccLwqlv5k7RPbxTqz9uI4FylLE,2781
-supervisely/nn/inference/tracking/tracker_interface.py,sha256=lrIo6ULa4GYd_QclALt82k5d9cLpH0Ss3--XFhPBgUQ,21913
+supervisely/nn/inference/tracking/tracker_interface.py,sha256=BrzddovTBxSsVt6eVa-v-6J7qYttK7S0JgJfd83M_eI,23070
 supervisely/nn/legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/nn/legacy/config.py,sha256=TKdyGtURJKzKoyydCZAfujoUnbC0SO8GeVLTSnoyS_w,2994
 supervisely/nn/legacy/dataset.py,sha256=-56EI6OYbkTWx4y8hOgD76y47zUoJNjGFyZ6JaP8iqg,6055
@@ -1083,9 +1083,9 @@ supervisely/worker_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 supervisely/worker_proto/worker_api_pb2.py,sha256=VQfi5JRBHs2pFCK1snec3JECgGnua3Xjqw_-b3aFxuM,59142
 supervisely/worker_proto/worker_api_pb2_grpc.py,sha256=3BwQXOaP9qpdi0Dt9EKG--Lm8KGN0C5AgmUfRv77_Jk,28940
 supervisely_lib/__init__.py,sha256=7-3QnN8Zf0wj8NCr2oJmqoQWMKKPKTECvjH9pd2S5vY,159
-supervisely-6.73.350.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.350.dist-info/METADATA,sha256=2hZYrEUmL3RyDgh_rG9RU5RdHEfNB7aRGMhQysvHAk4,33596
-supervisely-6.73.350.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-supervisely-6.73.350.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.350.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.350.dist-info/RECORD,,
+supervisely-6.73.352.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.352.dist-info/METADATA,sha256=Q3pBWWa63vwMk_BsKXdWp4bteVop0nTx0Gp0fOu6Uhg,33597
+supervisely-6.73.352.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+supervisely-6.73.352.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.352.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.352.dist-info/RECORD,,