jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-16 09:47:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats 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))