diff --git a/mediapipe/model_maker/python/vision/face_stylizer/dataset.py b/mediapipe/model_maker/python/vision/face_stylizer/dataset.py
index fd86df960..eb324028a 100644
--- a/mediapipe/model_maker/python/vision/face_stylizer/dataset.py
+++ b/mediapipe/model_maker/python/vision/face_stylizer/dataset.py
@@ -40,6 +40,12 @@ def _preprocess_face_dataset(
     tf.compat.v1.logging.info('Preprocess image %s', path)
     image = image_module.Image.create_from_file(path)
     aligned_image = aligner.align(image)
+    if aligned_image is None:
+      raise ValueError(
+          'ERROR: Invalid image. No face is detected and aligned. Please make'
+          ' sure the image has a single face that is facing straightforward and'
+          ' not significantly rotated.'
+      )
     aligned_image_tensor = tf.convert_to_tensor(aligned_image.numpy_view())
     preprocessed_images.append(aligned_image_tensor)
 
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py b/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py
index 914f50007..242140811 100644
--- a/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py
+++ b/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py
@@ -27,8 +27,8 @@ class DatasetTest(tf.test.TestCase):
 
   def test_from_image(self):
     test_image_file = 'input/style/cartoon/cartoon.jpg'
-    input_data_dir = test_utils.get_test_data_path(test_image_file)
-    data = dataset.Dataset.from_image(filename=input_data_dir)
+    input_image_path = test_utils.get_test_data_path(test_image_file)
+    data = dataset.Dataset.from_image(filename=input_image_path)
     self.assertEqual(data.num_classes, 1)
     self.assertEqual(data.label_names, ['cartoon'])
     self.assertLen(data, 1)
@@ -37,6 +37,12 @@ class DatasetTest(tf.test.TestCase):
     with self.assertRaisesRegex(ValueError, 'Unsupported image formats: .zip'):
       dataset.Dataset.from_image(filename='input/style/cartoon/cartoon.zip')
 
+  def test_from_image_raise_value_error_for_invalid_image(self):
+    with self.assertRaisesRegex(ValueError, 'Invalid image'):
+      test_image_file = 'input/style/sketch/boy-6030802_1280.jpg'
+      input_image_path = test_utils.get_test_data_path(test_image_file)
+      dataset.Dataset.from_image(filename=input_image_path)
+
 
 if __name__ == '__main__':
   tf.test.main()
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/boy-6030802_1280.jpg b/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/boy-6030802_1280.jpg
new file mode 100644
index 000000000..042f2c9d1
Binary files /dev/null and b/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/boy-6030802_1280.jpg differ