evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

catch key error when saving image (#18365)

Browse source
This commit is contained in:
Jason Hunter 2018-11-11 11:44:41 -05:00 • committed by Paulus Schoutsen
parent 5b9a9d8e04
commit 9cb6464c58
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
homeassistant/components/image_processing/tensorflow.py
View file

@ -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))