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

Internal change

Browse Source 
PiperOrigin-RevId: 489824381
This commit is contained in:
Scott Zhu 2022-11-20 10:39:59 -08:00 committed by Copybara-Service
parent bdf4078e89
commit a367753eda
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/model_maker/python/vision/gesture_recognizer/gesture_recognizer_test.py
View File

@ -14,7 +14,6 @@
import io import io
import os import os
import random
import tempfile import tempfile
from unittest import mock as unittest_mock from unittest import mock as unittest_mock
import zipfile import zipfile
@ -27,6 +26,7 @@ from mediapipe.model_maker.python.vision import gesture_recognizer
from mediapipe.tasks.python.test import test_utils from mediapipe.tasks.python.test import test_utils
_TEST_DATA_DIR = 'mediapipe/model_maker/python/vision/gesture_recognizer/test_data' _TEST_DATA_DIR = 'mediapipe/model_maker/python/vision/gesture_recognizer/test_data'
tf.keras.backend.experimental.enable_tf_random_generator()
class GestureRecognizerTest(tf.test.TestCase): class GestureRecognizerTest(tf.test.TestCase):
@ -42,7 +42,7 @@ class GestureRecognizerTest(tf.test.TestCase):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
random.seed(1234) tf.keras.utils.set_random_seed(87654321)
all_data = self._load_data() all_data = self._load_data()
# Splits data, 90% data for training, 10% for validation # Splits data, 90% data for training, 10% for validation
self._train_data, self._validation_data = all_data.split(0.9) self._train_data, self._validation_data = all_data.split(0.9)