Reorganize the face stylizer test data.
PiperOrigin-RevId: 536074239
This commit is contained in:
parent
2e2b4d183e
commit
759e9fd56e
|
@ -22,8 +22,7 @@ class DatasetTest(tf.test.TestCase):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
super().setUp()
|
super().setUp()
|
||||||
# TODO: Replace the stylize image dataset with licensed images.
|
self._test_data_dirname = 'input/style'
|
||||||
self._test_data_dirname = 'testdata'
|
|
||||||
|
|
||||||
def test_from_folder(self):
|
def test_from_folder(self):
|
||||||
input_data_dir = test_utils.get_test_data_path(self._test_data_dirname)
|
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):
|
def _load_data(self):
|
||||||
"""Loads training dataset."""
|
"""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)
|
data = face_stylizer.Dataset.from_folder(dirname=input_data_dir)
|
||||||
return data
|
return data
|
||||||
|
|
Binary file not shown.
Before Width: | Height: | Size: 347 KiB |
BIN
mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/cartoon/cartoon.jpg
vendored
Normal file
BIN
mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/cartoon/cartoon.jpg
vendored
Normal file
Binary file not shown.
After Width: | Height: | Size: 198 KiB |
BIN
mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/sketch.jpg
vendored
Normal file
BIN
mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/sketch.jpg
vendored
Normal file
Binary file not shown.
After Width: | Height: | Size: 185 KiB |
Binary file not shown.
Before Width: | Height: | Size: 336 KiB |
Loading…
Reference in New Issue
Block a user