From 9cb6464c587076e581233e3b009adc71837abc42 Mon Sep 17 00:00:00 2001 From: Jason Hunter Date: Sun, 11 Nov 2018 11:44:41 -0500 Subject: [PATCH] catch key error when saving image (#18365) --- homeassistant/components/image_processing/tensorflow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/image_processing/tensorflow.py b/homeassistant/components/image_processing/tensorflow.py index a2cd997bb76..2d06dbbcf34 100644 --- a/homeassistant/components/image_processing/tensorflow.py +++ b/homeassistant/components/image_processing/tensorflow.py @@ -246,7 +246,8 @@ class TensorFlowImageProcessor(ImageProcessingEntity): for category, values in matches.items(): # Draw custom category regions/areas - if self._category_areas[category] != [0, 0, 1, 1]: + if (category in self._category_areas + and self._category_areas[category] != [0, 0, 1, 1]): label = "{} Detection Area".format(category.capitalize()) draw_box(draw, self._category_areas[category], img_width, img_height, label, (0, 255, 0))