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

Added a test for min_gesture_confidence

Browse Source
This commit is contained in:
kinaryml 2022-10-28 01:38:15 -07:00
parent fbf7ba6f1a
commit b81b5a9035
3 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mediapipe/tasks/python/test/vision/gesture_recognizer_test.py
View File

@ -224,6 +224,24 @@ class GestureRecognizerTest(parameterized.TestCase):
self._assert_actual_result_approximately_matches_expected_result( self._assert_actual_result_approximately_matches_expected_result(
recognition_result, expected_recognition_result) recognition_result, expected_recognition_result)
def test_recognize_succeeds_with_min_gesture_confidence(self):
# Creates gesture recognizer.
base_options = _BaseOptions(model_asset_path=self.model_path)
options = _GestureRecognizerOptions(base_options=base_options,
min_gesture_confidence=2)
with _GestureRecognizer.create_from_options(options) as recognizer:
# Performs hand gesture recognition on the input.
recognition_result = recognizer.recognize(self.test_image)
expected_result = _get_expected_gesture_recognition_result(
_THUMB_UP_LANDMARKS, _THUMB_UP_LABEL, _THUMB_UP_INDEX)
# Only contains one top scoring gesture.
self.assertLen(recognition_result.gestures[0].classifications, 1)
# Actual gesture with top score matches expected gesture.
actual_top_gesture = recognition_result.gestures[0].classifications[0]
expected_top_gesture = expected_result.gestures[0].classifications[0]
self.assertEqual(actual_top_gesture.index, expected_top_gesture.index)
self.assertEqual(actual_top_gesture.label, expected_top_gesture.label)
def test_recognize_succeeds_with_num_hands(self): def test_recognize_succeeds_with_num_hands(self):
# Creates gesture recognizer. # Creates gesture recognizer.
base_options = _BaseOptions(model_asset_path=self.model_path) base_options = _BaseOptions(model_asset_path=self.model_path)

2
mediapipe/tasks/python/vision/core/image_processing_options.py
View File

@ -30,7 +30,7 @@ class ImageProcessingOptions:
Attributes: Attributes:
region_of_interest: The optional region-of-interest to crop from the image. region_of_interest: The optional region-of-interest to crop from the image.
If not specified, the full image is used. Coordinates must be in [0,1] If not specified, the full image is used. Coordinates must be in [0,1]
with 'left' < 'right' and 'top' < bottom. with 'x_center' < 'width' and 'y_center' < height.
rotation_degress: The rotation to apply to the image (or cropped rotation_degress: The rotation to apply to the image (or cropped
region-of-interest), in degrees clockwise. The rotation must be a region-of-interest), in degrees clockwise. The rotation must be a
multiple (positive or negative) of 90°. multiple (positive or negative) of 90°.

8
mediapipe/tasks/python/vision/gesture_recognizer.py
View File

@ -118,10 +118,10 @@ class GestureRecognizerOptions:
base_options: _BaseOptions base_options: _BaseOptions
running_mode: _RunningMode = _RunningMode.IMAGE running_mode: _RunningMode = _RunningMode.IMAGE
num_hands: Optional[int] = 1 num_hands: Optional[int] = 1
min_hand_detection_confidence: Optional[int] = 0.5 min_hand_detection_confidence: Optional[float] = 0.5
min_hand_presence_confidence: Optional[int] = 0.5 min_hand_presence_confidence: Optional[float] = 0.5
min_tracking_confidence: Optional[int] = 0.5 min_tracking_confidence: Optional[float] = 0.5
min_gesture_confidence: Optional[int] = -1 min_gesture_confidence: Optional[float] = -1
result_callback: Optional[ result_callback: Optional[
Callable[[GestureRecognitionResult, image_module.Image, Callable[[GestureRecognitionResult, image_module.Image,
int], None]] = None int], None]] = None