Source code for gbvision.utils.finders.rotated_rect_finder

from .object_finder import ShapeFinder
from gbvision.utils.shapes.base_rotated_rect import BaseRotatedRect
from ..shapes.base_shape import BaseShapeType


class RotatedRectFinder(ShapeFinder):
    """
    Finds a rectangular object, but rotated.
    Recommended to use when you know the shape isn't parallel to the camera
    """

    @staticmethod
    def _base_shape() -> BaseShapeType:
        return BaseRotatedRect