From 27052ec8bf62a2f18de16a55c886ec727563fb4e Mon Sep 17 00:00:00 2001
From: Nick Mowen <nickmowen213@gmail.com>
Date: Thu, 24 Aug 2023 08:26:57 -0600
Subject: [PATCH] Show sub label in debug view if available

---
 frigate/object_processing.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/frigate/object_processing.py b/frigate/object_processing.py
index 83684e1b6..f62dc3740 100644
--- a/frigate/object_processing.py
+++ b/frigate/object_processing.py
@@ -509,13 +509,14 @@ class CameraState:
 
                 # draw the bounding boxes on the frame
                 box = obj["box"]
+                text = obj["label"] if not obj.get("sub_label") else obj["sub_label"][0]
                 draw_box_with_label(
                     frame_copy,
                     box[0],
                     box[1],
                     box[2],
                     box[3],
-                    obj["label"],
+                    text,
                     f"{obj['score']:.0%} {int(obj['area'])}",
                     thickness=thickness,
                     color=color,