Reorganize the face stylizer test data.

PiperOrigin-RevId: 536074239
This commit is contained in:
MediaPipe Team 2023-05-28 19:25:33 -07:00 committed by Copybara-Service
parent 2e2b4d183e
commit 759e9fd56e
6 changed files with 2 additions and 3 deletions

View File

@ -22,8 +22,7 @@ class DatasetTest(tf.test.TestCase):
def setUp(self):
super().setUp()
# TODO: Replace the stylize image dataset with licensed images.
self._test_data_dirname = 'testdata'
self._test_data_dirname = 'input/style'
def test_from_folder(self):
input_data_dir = test_utils.get_test_data_path(self._test_data_dirname)

View File

@ -25,7 +25,7 @@ class FaceStylizerTest(tf.test.TestCase):
def _load_data(self):
"""Loads training dataset."""
input_data_dir = test_utils.get_test_data_path('testdata')
input_data_dir = test_utils.get_test_data_path('input/style')
data = face_stylizer.Dataset.from_folder(dirname=input_data_dir)
return data

Binary file not shown.

Before

Width:  |  Height:  |  Size: 347 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 198 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 185 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 336 KiB