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

internal change.

Browse Source 
PiperOrigin-RevId: 586156439
This commit is contained in:
MediaPipe Team 2023-11-28 18:05:07 -08:00 committed by Copybara-Service
parent 62bafd39bb
commit 91589b10d3
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mediapipe/python/solutions/drawing_utils.py
View File

@ -125,7 +125,8 @@ def draw_landmarks(
color=RED_COLOR), color=RED_COLOR),
connection_drawing_spec: Union[DrawingSpec, connection_drawing_spec: Union[DrawingSpec,
Mapping[Tuple[int, int], Mapping[Tuple[int, int],
DrawingSpec]] = DrawingSpec()): DrawingSpec]] = DrawingSpec(),
is_drawing_landmarks: bool = True):
"""Draws the landmarks and the connections on the image. """Draws the landmarks and the connections on the image.
Args: Args:
@ -142,6 +143,8 @@ def draw_landmarks(
connections to the DrawingSpecs that specifies the connections' drawing connections to the DrawingSpecs that specifies the connections' drawing
settings such as color and line thickness. If this argument is explicitly settings such as color and line thickness. If this argument is explicitly
set to None, no landmark connections will be drawn. set to None, no landmark connections will be drawn.
is_drawing_landmarks: Whether to draw landmarks. If set false, skip drawing
landmarks, only contours will be drawed.
Raises: Raises:
ValueError: If one of the followings: ValueError: If one of the followings:
@ -181,7 +184,7 @@ def draw_landmarks(
drawing_spec.thickness) drawing_spec.thickness)
# Draws landmark points after finishing the connection lines, which is # Draws landmark points after finishing the connection lines, which is
# aesthetically better. # aesthetically better.
if landmark_drawing_spec: if is_drawing_landmarks and landmark_drawing_spec:
for idx, landmark_px in idx_to_coordinates.items(): for idx, landmark_px in idx_to_coordinates.items():
drawing_spec = landmark_drawing_spec[idx] if isinstance( drawing_spec = landmark_drawing_spec[idx] if isinstance(
landmark_drawing_spec, Mapping) else landmark_drawing_spec landmark_drawing_spec, Mapping) else landmark_drawing_spec