cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added remaining parameters to initialize the Rect data class

Browse Source
This commit is contained in:
kinaryml 2022-10-10 13:58:37 -07:00
parent c2672d040f
commit 0a8dbc7576
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mediapipe/tasks/python/components/containers/rect.py
View File

@ -109,6 +109,8 @@ class NormalizedRect:
y_center=self.y_center, y_center=self.y_center,
width=self.width, width=self.width,
height=self.height, height=self.height,
rotation=self.rotation,
rect_id=self.rect_id
) )
@classmethod @classmethod
@ -119,7 +121,10 @@ class NormalizedRect:
x_center=pb2_obj.x_center, x_center=pb2_obj.x_center,
y_center=pb2_obj.y_center, y_center=pb2_obj.y_center,
width=pb2_obj.width, width=pb2_obj.width,
height=pb2_obj.height) height=pb2_obj.height,
rotation=pb2_obj.rotation,
rect_id=pb2_obj.rect_id
)
def __eq__(self, other: Any) -> bool: def __eq__(self, other: Any) -> bool:
"""Checks if this object is equal to the given object. """Checks if this object is equal to the given object.