From 6de275834deebd50c5c42a7ce2b5aba30d6f79ce Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Wed, 26 Jul 2023 17:54:59 -0700 Subject: [PATCH] internal change. PiperOrigin-RevId: 551366789 --- mediapipe/python/solutions/drawing_utils.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/mediapipe/python/solutions/drawing_utils.py b/mediapipe/python/solutions/drawing_utils.py index 1b8b173f7..ccad38a85 100644 --- a/mediapipe/python/solutions/drawing_utils.py +++ b/mediapipe/python/solutions/drawing_utils.py @@ -119,13 +119,16 @@ def draw_detection( def draw_landmarks( image: np.ndarray, landmark_list: landmark_pb2.NormalizedLandmarkList, - connections: Optional[List[Tuple[int, int]]] = None, - landmark_drawing_spec: Union[DrawingSpec, - Mapping[int, DrawingSpec]] = DrawingSpec( - color=RED_COLOR), - connection_drawing_spec: Union[DrawingSpec, - Mapping[Tuple[int, int], - DrawingSpec]] = DrawingSpec()): + connections: Optional[ + Union[frozenset[Tuple[int, int]], List[Tuple[int, int]]] + ] = None, + landmark_drawing_spec: Optional[Union[ + DrawingSpec, Mapping[int, DrawingSpec] + ]] = None, + connection_drawing_spec: Union[ + DrawingSpec, Mapping[Tuple[int, int], DrawingSpec] + ] = DrawingSpec(), +): """Draws the landmarks and the connections on the image. Args: