Merge pull request #308 from kuaashish/patch-1

LGTM. Update face_stylizer.ipynb
google-ai-edge · Jan 13, 2024 · 529e788 · 529e788
2 parents 20ff3fc + f0b1692
commit 529e788
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/customization/face_stylizer.ipynb b/examples/customization/face_stylizer.ipynb
@@ -264,7 +264,7 @@
         "iterator = iter(eval_output_data)\n",
         "\n",
         "reconstruct_style_image = (tf.squeeze(iterator.get_next()).numpy())\n",
-        "test_output_image = cv2.cvtColor(output_image, cv2.COLOR_RGB2BGR)\n",
+        "test_output_image = cv2.cvtColor(reconstruct_style_image, cv2.COLOR_RGB2BGR)\n",
         "print('\\nReconstructed style image')\n",
         "cv2_imshow(test_output_image)"
       ],