Added event_score_humand and event_score_vehicle Threshold setting fo…

…r Human / Vehicle classification to reduce false positives? #38
briis · Apr 18, 2022 · 773890d · 773890d
1 parent 96e0808
commit 773890d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/custom_components/securityspy/binary_sensor.py b/custom_components/securityspy/binary_sensor.py
@@ -18,7 +18,8 @@
 from .const import (
     ATTR_EVENT_LENGTH,
     ATTR_EVENT_OBJECT,
-    ATTR_EVENT_SCORE,
+    ATTR_EVENT_SCORE_HUMAN,
+    ATTR_EVENT_SCORE_VEHICLE,
     DOMAIN,
 )
 from .entity import SecuritySpyEntity
@@ -121,7 +122,8 @@ def extra_state_attributes(self):
                 ATTR_LAST_TRIP_TIME: self._device_data["last_motion"],
                 ATTR_EVENT_LENGTH: self._device_data["event_length"],
                 ATTR_EVENT_OBJECT: self._device_data["event_object"],
-                ATTR_EVENT_SCORE: self._device_data["event_score"],
+                ATTR_EVENT_SCORE_HUMAN: self._device_data["event_score_human"],
+                ATTR_EVENT_SCORE_VEHICLE: self._device_data["event_score_vehicle"],
             }
         return {
             **super().extra_state_attributes,

diff --git a/custom_components/securityspy/const.py b/custom_components/securityspy/const.py
@@ -29,7 +29,8 @@
 ATTR_BRAND = "brand"
 ATTR_EVENT_LENGTH = "event_length"
 ATTR_EVENT_OBJECT = "event_object"
-ATTR_EVENT_SCORE = "event_score"
+ATTR_EVENT_SCORE_HUMAN = "event_score_human"
+ATTR_EVENT_SCORE_VEHICLE = "event_score_vehicle"
 ATTR_PRESET_ID = "preset_id"
 ATTR_PTZ_CAPABILITIES = "ptz_capabilities"
 

diff --git a/custom_components/securityspy/sensor.py b/custom_components/securityspy/sensor.py
@@ -14,7 +14,8 @@
 from .entity import SecuritySpyEntity
 
 from .const import (
-    ATTR_EVENT_SCORE,
+    ATTR_EVENT_SCORE_HUMAN,
+    ATTR_EVENT_SCORE_VEHICLE,
     DEVICE_CLASS_DETECTION,
     DOMAIN,
     RECORDING_TYPE_ACTION,
@@ -130,7 +131,8 @@ def extra_state_attributes(self):
         if self._description.device_type == DEVICE_CLASS_DETECTION:
             return {
                 **super().extra_state_attributes,
-                ATTR_EVENT_SCORE: self._device_data["event_score"],
+                ATTR_EVENT_SCORE_HUMAN: self._device_data["event_score_human"],
+                ATTR_EVENT_SCORE_VEHICLE: self._device_data["event_score_vehicle"],
             }
         return {
             **super().extra_state_attributes,