Update face_stylizer.ipynb

output "output_mage" variable is not declared. rather the result is stored in "reconstruct_style_image"
google-ai-edge · Jan 10, 2024 · f0b1692 · f0b1692
1 parent 20ff3fc
commit f0b1692
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)"
       ],