6
0

MediaBoundingBox.py 451 B

1234567891011121314
  1. from pycs.database.Result import Result
  2. class MediaBoundingBox:
  3. def __init__(self, result: Result):
  4. self.x = result.data['x']
  5. self.y = result.data['y']
  6. self.w = result.data['w']
  7. self.h = result.data['h']
  8. self.label = result.label
  9. self.frame = result.data['frame'] if 'frame' in result.data else None
  10. def serialize(self) -> dict:
  11. return dict({'type': 'bounding-box'}, **self.__dict__)