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

Internal Changes

Browse Source 
PiperOrigin-RevId: 522247775
This commit is contained in:
MediaPipe Team 2023-04-05 22:29:23 -07:00 committed by Copybara-Service
parent 7455022980
commit 5a1a9269e6
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/model_maker/python/core/utils/file_util.py
View File

@ -94,4 +94,6 @@ class DownloadedFiles:
pathlib.Path.mkdir(absolute_path.parent, parents=True, exist_ok=True) pathlib.Path.mkdir(absolute_path.parent, parents=True, exist_ok=True)
with open(absolute_path, 'wb') as f: with open(absolute_path, 'wb') as f:
f.write(r.content) f.write(r.content)
else:
print(f'Using existing files at {absolute_path}')
return str(absolute_path) return str(absolute_path)

10
mediapipe/model_maker/python/vision/gesture_recognizer/gesture_recognizer_test.py
View File

@ -15,7 +15,6 @@
import io import io
import os import os
import tempfile import tempfile
import unittest
from unittest import mock as unittest_mock from unittest import mock as unittest_mock
import zipfile import zipfile
@ -32,7 +31,6 @@ _TEST_DATA_DIR = 'mediapipe/model_maker/python/vision/gesture_recognizer/testdat
tf.keras.backend.experimental.enable_tf_random_generator() tf.keras.backend.experimental.enable_tf_random_generator()
@unittest.skip('b/273818271')
class GestureRecognizerTest(tf.test.TestCase): class GestureRecognizerTest(tf.test.TestCase):
def _load_data(self): def _load_data(self):
@ -47,9 +45,6 @@ class GestureRecognizerTest(tf.test.TestCase):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
tf.keras.utils.set_random_seed(87654321) tf.keras.utils.set_random_seed(87654321)
all_data = self._load_data()
# Splits data, 90% data for training, 10% for validation
self._train_data, self._validation_data = all_data.split(0.9)
# Mock tempfile.gettempdir() to be unique for each test to avoid race # Mock tempfile.gettempdir() to be unique for each test to avoid race
# condition when downloading model since these tests may run in parallel. # condition when downloading model since these tests may run in parallel.
mock_gettempdir = unittest_mock.patch.object( mock_gettempdir = unittest_mock.patch.object(
@ -60,6 +55,10 @@ class GestureRecognizerTest(tf.test.TestCase):
) )
self.mock_gettempdir = mock_gettempdir.start() self.mock_gettempdir = mock_gettempdir.start()
self.addCleanup(mock_gettempdir.stop) self.addCleanup(mock_gettempdir.stop)
# Load dataset used by tests
all_data = self._load_data()
# Splits data, 90% data for training, 10% for validation
self._train_data, self._validation_data = all_data.split(0.9)
def test_gesture_recognizer_model(self): def test_gesture_recognizer_model(self):
mo = gesture_recognizer.ModelOptions() mo = gesture_recognizer.ModelOptions()
@ -74,7 +73,6 @@ class GestureRecognizerTest(tf.test.TestCase):
self._test_accuracy(model) self._test_accuracy(model)
@unittest.skip('b/273818271')
@unittest_mock.patch.object( @unittest_mock.patch.object(
tf.keras.layers, 'Dense', wraps=tf.keras.layers.Dense tf.keras.layers, 'Dense', wraps=tf.keras.layers.Dense
) )