147 lines
6.1 KiB
Python
147 lines
6.1 KiB
Python
# Copyright 2021 The MediaPipe Authors.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless requi_RED by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""MediaPipe solution drawing styles."""
|
|
|
|
from typing import Mapping, Tuple
|
|
|
|
from mediapipe.python.solutions.drawing_utils import DrawingSpec
|
|
from mediapipe.python.solutions.hands import HandLandmark
|
|
|
|
_RADIUS = 5
|
|
_RED = (54, 67, 244)
|
|
_GREEN = (118, 230, 0)
|
|
_BLUE = (192, 101, 21)
|
|
_YELLOW = (0, 204, 255)
|
|
_GRAY = (174, 164, 144)
|
|
_PURPLE = (128, 64, 128)
|
|
_PEACH = (180, 229, 255)
|
|
|
|
# Hands
|
|
_THICKNESS_WRIST_MCP = 3
|
|
_THICKNESS_FINGER = 2
|
|
_THICKNESS_DOT = -1
|
|
|
|
# Hand landmarks
|
|
_PALM_LANMARKS = (HandLandmark.WRIST, HandLandmark.THUMB_CMC,
|
|
HandLandmark.INDEX_FINGER_MCP, HandLandmark.MIDDLE_FINGER_MCP,
|
|
HandLandmark.RING_FINGER_MCP, HandLandmark.PINKY_MCP)
|
|
_THUMP_LANDMARKS = (HandLandmark.THUMB_MCP, HandLandmark.THUMB_IP,
|
|
HandLandmark.THUMB_TIP)
|
|
_INDEX_FINGER_LANDMARKS = (HandLandmark.INDEX_FINGER_PIP,
|
|
HandLandmark.INDEX_FINGER_DIP,
|
|
HandLandmark.INDEX_FINGER_TIP)
|
|
_MIDDLE_FINGER_LANDMARKS = (HandLandmark.MIDDLE_FINGER_PIP,
|
|
HandLandmark.MIDDLE_FINGER_DIP,
|
|
HandLandmark.MIDDLE_FINGER_TIP)
|
|
_RING_FINGER_LANDMARKS = (HandLandmark.RING_FINGER_PIP,
|
|
HandLandmark.RING_FINGER_DIP,
|
|
HandLandmark.RING_FINGER_TIP)
|
|
_PINKY_FINGER_LANDMARKS = (HandLandmark.PINKY_PIP, HandLandmark.PINKY_DIP,
|
|
HandLandmark.PINKY_TIP)
|
|
_HAND_LANDMARK_STYLE = {
|
|
_PALM_LANMARKS:
|
|
DrawingSpec(
|
|
color=_RED, thickness=_THICKNESS_DOT, circle_radius=_RADIUS),
|
|
_THUMP_LANDMARKS:
|
|
DrawingSpec(
|
|
color=_PEACH, thickness=_THICKNESS_DOT, circle_radius=_RADIUS),
|
|
_INDEX_FINGER_LANDMARKS:
|
|
DrawingSpec(
|
|
color=_PURPLE, thickness=_THICKNESS_DOT, circle_radius=_RADIUS),
|
|
_MIDDLE_FINGER_LANDMARKS:
|
|
DrawingSpec(
|
|
color=_YELLOW, thickness=_THICKNESS_DOT, circle_radius=_RADIUS),
|
|
_RING_FINGER_LANDMARKS:
|
|
DrawingSpec(
|
|
color=_GREEN, thickness=_THICKNESS_DOT, circle_radius=_RADIUS),
|
|
_PINKY_FINGER_LANDMARKS:
|
|
DrawingSpec(
|
|
color=_BLUE, thickness=_THICKNESS_DOT, circle_radius=_RADIUS),
|
|
}
|
|
|
|
# Hand connections
|
|
_PALM_CONNECTIONS = ((HandLandmark.WRIST, HandLandmark.THUMB_CMC),
|
|
(HandLandmark.WRIST, HandLandmark.INDEX_FINGER_MCP),
|
|
(HandLandmark.MIDDLE_FINGER_MCP,
|
|
HandLandmark.RING_FINGER_MCP),
|
|
(HandLandmark.RING_FINGER_MCP, HandLandmark.PINKY_MCP),
|
|
(HandLandmark.INDEX_FINGER_MCP,
|
|
HandLandmark.MIDDLE_FINGER_MCP), (HandLandmark.WRIST,
|
|
HandLandmark.PINKY_MCP))
|
|
_THUMB_CONNECTIONS = ((HandLandmark.THUMB_CMC, HandLandmark.THUMB_MCP),
|
|
(HandLandmark.THUMB_MCP, HandLandmark.THUMB_IP),
|
|
(HandLandmark.THUMB_IP, HandLandmark.THUMB_TIP))
|
|
_INDEX_FINGER_CONNECTIONS = ((HandLandmark.INDEX_FINGER_MCP,
|
|
HandLandmark.INDEX_FINGER_PIP),
|
|
(HandLandmark.INDEX_FINGER_PIP,
|
|
HandLandmark.INDEX_FINGER_DIP),
|
|
(HandLandmark.INDEX_FINGER_DIP,
|
|
HandLandmark.INDEX_FINGER_TIP))
|
|
_MIDDLE_FINGER_CONNECTIONS = ((HandLandmark.MIDDLE_FINGER_MCP,
|
|
HandLandmark.MIDDLE_FINGER_PIP),
|
|
(HandLandmark.MIDDLE_FINGER_PIP,
|
|
HandLandmark.MIDDLE_FINGER_DIP),
|
|
(HandLandmark.MIDDLE_FINGER_DIP,
|
|
HandLandmark.MIDDLE_FINGER_TIP))
|
|
_RING_FINGER_CONNECTIONS = ((HandLandmark.RING_FINGER_MCP,
|
|
HandLandmark.RING_FINGER_PIP),
|
|
(HandLandmark.RING_FINGER_PIP,
|
|
HandLandmark.RING_FINGER_DIP),
|
|
(HandLandmark.RING_FINGER_DIP,
|
|
HandLandmark.RING_FINGER_TIP))
|
|
_PINKY_FINGER_CONNECTIONS = ((HandLandmark.PINKY_MCP, HandLandmark.PINKY_PIP),
|
|
(HandLandmark.PINKY_PIP, HandLandmark.PINKY_DIP),
|
|
(HandLandmark.PINKY_DIP, HandLandmark.PINKY_TIP))
|
|
_HAND_CONNECTION_STYLE = {
|
|
_PALM_CONNECTIONS:
|
|
DrawingSpec(color=_GRAY, thickness=_THICKNESS_WRIST_MCP),
|
|
_THUMB_CONNECTIONS:
|
|
DrawingSpec(color=_PEACH, thickness=_THICKNESS_FINGER),
|
|
_INDEX_FINGER_CONNECTIONS:
|
|
DrawingSpec(color=_PURPLE, thickness=_THICKNESS_FINGER),
|
|
_MIDDLE_FINGER_CONNECTIONS:
|
|
DrawingSpec(color=_YELLOW, thickness=_THICKNESS_FINGER),
|
|
_RING_FINGER_CONNECTIONS:
|
|
DrawingSpec(color=_GREEN, thickness=_THICKNESS_FINGER),
|
|
_PINKY_FINGER_CONNECTIONS:
|
|
DrawingSpec(color=_BLUE, thickness=_THICKNESS_FINGER)
|
|
}
|
|
|
|
|
|
def get_default_hand_landmark_style() -> Mapping[int, DrawingSpec]:
|
|
"""Returns the default hand landmark drawing style.
|
|
|
|
Returns:
|
|
A mapping from each hand landmark to the default drawing spec.
|
|
"""
|
|
hand_landmark_style = {}
|
|
for k, v in _HAND_LANDMARK_STYLE.items():
|
|
for landmark in k:
|
|
hand_landmark_style[landmark] = v
|
|
return hand_landmark_style
|
|
|
|
|
|
def get_default_hand_connection_style(
|
|
) -> Mapping[Tuple[int, int], DrawingSpec]:
|
|
"""Returns the default hand connection drawing style.
|
|
|
|
Returns:
|
|
A mapping from each hand connection to the default drawing spec.
|
|
"""
|
|
hand_connection_style = {}
|
|
for k, v in _HAND_CONNECTION_STYLE.items():
|
|
for connection in k:
|
|
hand_connection_style[connection] = v
|
|
return hand_connection_style
|