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

Removed min score thres from tests

Browse Source
This commit is contained in:
kinaryml 2022-10-31 23:07:05 -07:00
parent d3b472e888
commit a913255080
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -367,7 +367,7 @@ class GestureRecognizerTest(parameterized.TestCase):
_GESTURE_RECOGNIZER_WITH_CUSTOM_CLASSIFIER_BUNDLE_ASSET_FILE) _GESTURE_RECOGNIZER_WITH_CUSTOM_CLASSIFIER_BUNDLE_ASSET_FILE)
base_options = _BaseOptions(model_asset_path=model_path) base_options = _BaseOptions(model_asset_path=model_path)
canned_gesture_classifier_options = _ClassifierOptions( canned_gesture_classifier_options = _ClassifierOptions(
score_threshold=.5, category_allowlist=[ category_allowlist=[
'None', 'Open_Palm', 'Victory', 'Thumb_Down', 'Thumb_Up', 'None', 'Open_Palm', 'Victory', 'Thumb_Down', 'Thumb_Up',
'ILoveYou', 'Closed_Fist']) 'ILoveYou', 'Closed_Fist'])
options = _GestureRecognizerOptions( options = _GestureRecognizerOptions(
@ -389,7 +389,7 @@ class GestureRecognizerTest(parameterized.TestCase):
_GESTURE_RECOGNIZER_WITH_CUSTOM_CLASSIFIER_BUNDLE_ASSET_FILE) _GESTURE_RECOGNIZER_WITH_CUSTOM_CLASSIFIER_BUNDLE_ASSET_FILE)
base_options = _BaseOptions(model_asset_path=model_path) base_options = _BaseOptions(model_asset_path=model_path)
canned_gesture_classifier_options = _ClassifierOptions( canned_gesture_classifier_options = _ClassifierOptions(
score_threshold=.5, category_allowlist=['Pointing_Up'], category_allowlist=['Pointing_Up'],
category_denylist=['Pointing_Up']) category_denylist=['Pointing_Up'])
options = _GestureRecognizerOptions( options = _GestureRecognizerOptions(
base_options=base_options, base_options=base_options,