Reorganize the face stylizer test data.
PiperOrigin-RevId: 536074239
This commit is contained in:
committed by
Copybara-Service
parent
2e2b4d183e
commit
759e9fd56e
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
BIN
Binary file not shown.
|
Before Width: | Height: | Size: 347 KiB |
BIN
Binary file not shown.
|
After Width: | Height: | Size: 198 KiB |
BIN
Binary file not shown.
|
After Width: | Height: | Size: 185 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 336 KiB |
Reference in New Issue
Block a user