From ca337c32b4abb7bd66341de984a946241559d2dd Mon Sep 17 00:00:00 2001 From: Blake Blackshear Date: Mon, 21 Dec 2020 08:03:27 -0600 Subject: [PATCH] better mask error handling --- frigate/config.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/frigate/config.py b/frigate/config.py index f9b769f96..aaf53d746 100644 --- a/frigate/config.py +++ b/frigate/config.py @@ -1,5 +1,6 @@ import base64 import json +import logging import os from typing import Dict @@ -11,6 +12,8 @@ import yaml from frigate.const import RECORD_DIR, CLIPS_DIR, CACHE_DIR +logger = logging.getLogger(__name__) + DETECTORS_SCHEMA = vol.Schema( { vol.Required(str): { @@ -743,7 +746,9 @@ class CameraConfig(): cv2.fillPoly(mask_img, pts=[contour], color=(0)) else: mask_file = cv2.imread(f"/config/{mask}", cv2.IMREAD_GRAYSCALE) - if not mask_file.size == 0: + if mask_file is None or mask_file.size == 0: + logger.warning(f"Could not read mask file {mask}") + else: mask_img[np.where(mask_file==[0])] = [0] def _get_ffmpeg_cmd(self, ffmpeg_input):