sublabel score

ljmerza · Mar 10, 2024 · fe45130 · fe45130
1 parent b876ecd
commit fe45130
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -18,6 +18,7 @@ frigate:
   mqtt_username: username
   mqtt_password: password
   main_topic: frigate
+  sublabel_score: true
   camera:
     - birdcam
 bird_classification:

diff --git a/index.py b/index.py
@@ -28,7 +28,7 @@
 firstmessage = True
 _LOGGER = None
 
-VERSION = '1.0.4'
+VERSION = '1.0.5'
 
 CONFIG_PATH = './config/config.yml'
 DB_PATH = './config/classifier.db'
@@ -273,6 +273,10 @@ def on_message(client, userdata, message):
         _LOGGER.error(f"Unknown label: {after_data['label']}")
         return
 
+    # add score to sublabel
+    if config['frigate'].get('sublabel_score'):
+        sublabel = f'{sublabel} ({int(score * 100)}%)'
+
     if result is None:
         # Insert a new record if it doesn't exist
         _LOGGER.info("No record yet for this event. Storing.")

diff --git a/requirements.txt b/requirements.txt
@@ -4,6 +4,4 @@ paho_mqtt==1.6.1
 PyYAML==6.0.1
 tflite-support==0.4.4
 requests==2.31.0
-Pillow==10.0.0
-tf-nightly==2.15.0.dev20230803
-scipy==1.11.1
+Pillow==10.0.0