base_video_result.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. # copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. from .base_result import BaseResult
  15. from .mixin import StrMixin, JsonMixin, ImgMixin, VideoMixin
  16. from ...utils.io import VideoReader, VideoWriter
  17. class BaseVideoResult(BaseResult, StrMixin, JsonMixin, VideoMixin):
  18. """Base class for computer vision results."""
  19. INPUT_IMG_KEY = "input_img"
  20. def __init__(self, data: dict) -> None:
  21. """
  22. Initialize the BaseVideoResult.
  23. Args:
  24. data (dict): The initial data.
  25. Raises:
  26. AssertionError: If the required key (`BaseVideoResult.INPUT_IMG_KEY`) are not found in the data.
  27. """
  28. self._video_reader = VideoReader(backend="decord")
  29. self._video_writer = VideoWriter(backend="opencv")
  30. super().__init__(data)
  31. StrMixin.__init__(self)
  32. JsonMixin.__init__(self)
  33. VideoMixin.__init__(self, "opencv")