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